다음 그림과 같이 지뢰가 있는 지역과 지뢰에 인접한 위, 아래, 좌, 우 대각선 칸을 모두 위험지역으로 분류합니다.
지뢰는 2차원 배열 board에 1로 표시되어 있고, board에는 지뢰가 매설 된 지역 1과, 지뢰가 없는 지역 0만 존재합니다.
지뢰가 매설된 지역의 지도 board가 매개변수로 주어질 때, 안전한 지역의 칸 수를 return하도록 solution 함수를 완성해주세요.
board | result |
---|---|
[[0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 1, 0, 0], [0, 0, 0, 0, 0]] | 16 |
[[0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 0, 0, 0], [0, 0, 1, 1, 0], [0, 0, 0, 0, 0]] | 13 |
[[1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1, 1], [1, 1, 1, 1, 1, 1]] | 0 |
입출력 예 설명
1. n = board의 길이
2. for x in range(n) and for y in range(n): 좌표
3. (x,y)에 지뢰가 있다면:
4. for (r,c) in 주변 8칸:
5. (r,c)가 board 안이며 안전지역이라면:
6. 위험지역(2)로 지정
7. 안전지역(0)인 칸의 개수 세기
import itertools
def solution(board):
n = len(board)
for x, y in list(itertools.product(range(n), repeat=2)):
if board[x][y]==1: # 지뢰가 있을 때
for i, j in list(itertools.product((-1, 0, 1), repeat=2)):# 방향
r, c = x+i, y+j
# borad 안이고 0일 때
if -1 < r < n and -1 < c < n and board[r][c] == 0:
board[r][c] = 2
# print(board)
return sum([1 for r in board for c in r if not c]) # 안전지역 칸 수
enumerate
함수를 사용해 좌표의 위치와 칸의 정보를 모두 접근한다.update
한다.def solution(board):
n = len(board)
danger = set()
for i, row in enumerate(board):
for j, x in enumerate(row):
if not x:
continue
danger.update((i+di, j+dj) for di in [-1,0,1] for dj in [-1, 0, 1])
return n*n - sum(0 <= i < n and 0 <= j < n for i, j in danger)