(Python)백준 코딩테스트 연습 - 토마토(7569)

1. 문제

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

7569번: 토마토

2. 풀이

# 7569번 : 토마토
# 입력 코드
import sys
input = sys.stdin.readline
from collections import deque

# m : 상자 가로 칸 수
# n : 상자 세로 칸 수
# h : 쌓는 상자의 수
m, n, h = map(int, input().split())

# boxList : 박스에 담긴 토마토의 정보 배열
boxList = [[list(map(int, input().split()))for j in range(n)]for i in range(h)]

# 최초 입력받은 토마토 정보 중 익은 토마토의 좌표를 찾는 코드
q = deque()

for i in range(h):
    for j in range(n):
        for k in range(m):
            if boxList[i][j][k] == 1:
                q.append([i, j, k])
                
# 익은 토마토가 안익은 토마토를 익히는 코드
cnt = 0
while q:
    new_q = deque()
    cnt += 1
    while q:
        x, y, z = map(int, q.popleft())
        if x - 1 >= 0 and boxList[x-1][y][z] == 0:
            boxList[x-1][y][z] = 1
            new_q.append([x-1, y, z])
        if x + 1 < h and boxList[x+1][y][z] == 0:
            boxList[x+1][y][z] = 1
            new_q.append([x+1, y, z])
        if y - 1 >= 0 and boxList[x][y-1][z] == 0:
            boxList[x][y-1][z] = 1
            new_q.append([x, y-1, z])
        if y + 1 < n and boxList[x][y+1][z] == 0:
            boxList[x][y+1][z] = 1
            new_q.append([x, y+1, z])
        if z - 1 >= 0 and boxList[x][y][z-1] == 0:
            boxList[x][y][z-1] = 1
            new_q.append([x, y, z-1])
        if z + 1 < m and boxList[x][y][z+1] == 0:
            boxList[x][y][z+1] = 1
            new_q.append([x, y, z+1])
    q = new_q

containZero = False
for i in boxList:
    for j in i:
        if 0 in j:
            containZero = True
            break
    if containZero: break
        
if containZero:
    print(-1)
else:
    print(cnt - 1)