백준7576(파이썬) - 토마토

https://www.acmicpc.net/problem/7576

7576번: 토마토

• 이 문제는 보자마자 BFS라고 생각을 했다.
• 모든 토마토가 익을 때까지의 시간을 구하는게 핵심이므로 BFS를 한번 진행할때마다 카운트를 해줘야겠다.
• '저장될 때부터 모든 토마토가 익어있는 상태이면 0을 출력해야 하고, 토마토가 모두 익지는 못하는 상황이면 -1을 출력해야 한다.' 라는 문제 지문에서 0과 -1과 결과값을 어떻게 구별해서 출력할지를 생각했다.
• 보통 재귀를 사용했었는데, dx,dy와 queue를 사용해서 DFS 또는 BFS를 구현하는 연습을 하고있다.

import sys
from collections import deque
input = sys.stdin.readline
m,n = map(int,input().split())

data = []
dx = [-1,1,0,0]
dy = [0,0,-1,1]
for _ in range(n):
    data.append(list(map(int,input().split())))

queue = deque()
def bfs():
    while queue:
        x,y = queue.popleft()
        for i in range(4):
            nx = x + dx[i]
            ny = y + dy[i]
            if nx>-1 and nx<n and ny>-1 and ny<m:
                if data[nx][ny] == 0:
                    data[nx][ny] = data[x][y]+1
                    queue.append((nx,ny))
            
for i in range(n):
    for j in range(m):
        if data[i][j] == 1:
            queue.append((i,j))
bfs()
flag = 0
result = -2
for i in range(n):
    for j in range(m):
        if data[i][j] == 0:
            flag = 1
        result = max(result,data[i][j])
if flag == 1:
    print(-1)
elif result == -1:
    print(0)
else:
    print(result-1)