안전 거리의 최댓값 구하기.
입력 | 출력 |
---|---|
5 4 0 0 1 0 0 0 0 0 1 0 0 0 0 0 0 0 0 0 0 1 | 2 |
7 4 0 0 0 1 0 1 0 0 0 0 0 0 0 0 0 1 0 0 0 0 0 1 0 0 0 0 0 1 | 2 |
: bfs사용. 모든 1에서 탐색 어떻게?
전에 풀어본 백준 토마토 문제와 같다.
입력 받은 공간의 상태 값이 1인 좌표를 큐에 넣고 bfs진행.
visited의 max값 출력.
from collections import deque
def bfs():
dx = [-1,1,0,0,-1,-1,1,1]
dy = [0,0,-1,1,-1,1,-1,1]
while q:
x, y = q.popleft()
for k in range(8):
nx = x + dx[k]
ny = y + dy[k]
if 0<=nx<n and 0<=ny<m:
if maps[nx][ny] == 0 and visited[nx][ny]==0::
visited[nx][ny] = visited[x][y]+1
q.append((nx, ny))
n, m = map(int, input().split())
maps = [list(map(int, input().split())) for _ in range(n)]
visited = [[0]*m for _ in range(n)]
q = deque()
for i in range(n):
for j in range(m):
if maps[i][j] == 1:
q.append((i, j))
bfs()
print(max(map(max, visited)))