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

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

7569번: 토마토

• 7576번 토마토 문제와 같은 방식이지만 7576번 처럼 2차원배열이 아닌 3차원 배열로 해결해야 하는 문제였다.
• x,y,z축으로 생각을 해주고 문제를 풀어주는 방식은 같다.
• 내가 삽질을 했던 부분은 높이를 z로 두었을 때, x,y,z 순서로 for문을 돌리는 것이 아닌 z,x,y로 돌려야 한다는 것이다.
• 나머지는 7576 토마토 문제와 같다. 풀이는 아래 게시물을 참고하였다.

2021.05.20 - [Algorithm & Python/백준] - 백준7576(파이썬) - 토마토

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

 

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

m,n,h = map(int,input().split())


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

data = [[list(map(int, input().split())) for _ in range(n)] for _ in range(h)]
queue = deque()

#3차원 bfs문제
def bfs():
    while queue:
        # 높이, x,y 순서
        z,x,y = queue.popleft()
        for i in range(6):
            nx = x + dx[i]
            ny = y + dy[i]
            nz = z + dz[i]
            if -1<nx<n and -1<ny<m and -1<nz<h:
                # 높이, x,y 순서
                if data[nz][nx][ny] == 0:
                    data[nz][nx][ny] = data[z][x][y]+1
                    queue.append((nz,nx,ny))
            
for i in range(h):
    for j in range(n):
        for k in range(m):
            # 높이, x,y 순서
            if data[i][j][k] == 1:
                # 높이, x,y 순서
                queue.append((i,j,k))
bfs()
flag = 0
result = -2
for i in range(h):
    for j in range(n):
        for k in range(m):
            # 높이, x,y 순서
            if data[i][j][k] == 0:
                flag = 1
                # 높이, x,y 순서
            result = max(result,data[i][j][k])
if flag == 1:
    print(-1)
elif result == -1:
    print(0)
else:
    print(result-1)