[BOJ] Python 백준 토마토(7569)

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


기존에 2차원 토마토 문제에서 더 나아간 3차원 토마토이다. 초기 토마토 위치를 파악해 큐에 넣고 bfs를 통해 전염 시켜준다. 이를 통해 전염이 다 되었을때 증가하는 날짜를 확인한다.

import sys
input = sys.stdin.readline
from collections import deque

N, M, H = map(int,input().split())

s = [[list(map(int,input().split())) for _ in range(M)] for _ in range(H)]

di = [-1, 1, 0, 0, 0, 0]
dj = [0, 0, -1, 1, 0, 0]
dz = [0, 0, 0, 0, -1, 1]


queue = deque()

def bfs():
    while queue:
        cz,ci,cj = queue.popleft()
        for i in range(6):
            nz, ni, nj = cz + dz[i], ci + di[i], cj + dj[i]
            if 0<= nz < H and 0 <= ni < M and 0 <= nj < N:
                if s[nz][ni][nj] == 0:
                    s[nz][ni][nj] = s[cz][ci][cj] + 1
                    queue.append((nz,ni,nj))


for z in range(H):
    for i in range(M):
        for j in range(N):
            if s[z][i][j] == 1:
                queue.append((z,i,j))


bfs()
day = 0
for z in range(H):
    for i in range(M):
        for j in range(N):
            if s[z][i][j] == 0:
                print(-1)
                exit()
            day = max(day, s[z][i][j])

print(day-1)

회고,
queue를 전역으로 선언하지 않고 bfs에 큐를 각자 생성해서 조정하는 부분에서 꽤 헤맸던것 같다....
다양하게 bfs를 사용하는법을 해봐야겠다.