안전지대

프로그래머스
정답률: 58%
영역:탐색
나의 예상 정답률:63.5%
상승률:1.7%

지뢰는 2차원 배열 board에 1로 표시되어 있고 board에는 지뢰가 매설 된 지역 1과, 지뢰가 없는 지역 0만 존재합니다.
지뢰가 매설된 지역의 지도 board가 매개변수로 주어질 때, 안전한 지역의 칸 수를 return하도록 solution 함수를 완성해주세요.
board는 n * n 배열입니다.
1 ≤ n ≤ 100
지뢰는 1로 표시되어 있습니다.
board에는 지뢰가 있는 지역 1과 지뢰가 없는 지역 0만 존재합니다.

# input setting
problem_num = '안전지대'

import os, sys
path = os.getcwd() + f'\\txt\\{problem_num}'  + '.txt'
sys.stdin = open(path,'r')
board = sys.stdin.readlines()
board = [list(map(int,s.strip().split(' '))) for s in board]

import numpy as np
def solution(board):
    mask = np.zeros_like(board)
    i,j = mask.shape[0],mask.shape[1]
    for n in range(i):
        for m in range(j):
            if board[n][m] == 1:
                for a in range(n-1 if n-1 > 0 else 0, n+2 if n+2 < i else i):
                    for b in range(m-1 if m-1 > 0 else 0, m+2 if m+2 < j else j):
                        mask[a][b] = 1
    answer = int(i*j-mask.sum())
    return answer

solution(board)

13

• 마스킹

• np.sum for ndarray

• map의 iterable 요소를 풀어서 넣으면 [map~]이렇게 넣으면 안되고 list(map~) 이렇게 넣어줘야한다.