[백준-7569] 토마토 / Python

7569번: 토마토

from collections import deque
import sys

m, n, h = map(int, sys.stdin.readline().split())

dx = [-1, 1, 0, 0, 0, 0]
dy = [0, 0, -1, 1, 0, 0]
dz = [0, 0, 0, 0, -1, 1]


def bfs():
    while queue:
        x, y, z = queue.popleft()
        for i in range(6):
            nx = x + dx[i]
            ny = y + dy[i]
            nz = z + dz[i]
            if 0 <= nx < n and 0 <= ny < m and 0 <= nz < h and graph[nz][nx][ny] == 0:
                queue.append([nx, ny, nz])
                graph[nz][nx][ny] = graph[z][x][y] + 1


graph = [[] for _ in range(h)]
queue = deque()
isTrue = False
for i in range(h):
    for j in range(n):
        graph[i].append(list(map(int, sys.stdin.readline().split())))

for z in range(h):
    for x in range(n):
        for y in range(m):
            if graph[z][x][y] == 1:
                queue.append([x, y, z])

bfs()

maxValue = 0
for z in range(h):
    for x in range(n):
        for y in range(m):
            if graph[z][x][y] == 0:
                isTrue = True
                break
            maxValue = max(maxValue, graph[z][x][y])

if isTrue == True:
    print(-1)
else:
    print(maxValue - 1)

 

해설

이전에 풀었던 토마토 문제와 비슷한 유형이다. 다만 이 문제에서는 높이가 추가됐다.

가로, 세로, 높이에 맞게 익은 토마토를 확인하고 deque에 넣어서 bfs를 진행했다.