BOJ 7569 토마토(Python 3)

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

바로 이전 글에 있는 토마토 문제의 3차원 버전 문제이다.

여기에서 주의할 점은 M,N,H로 주어지는 값이 배열에서는 List[H][N][M] 이다.

이 부분만 유의해서 만든다면 2차원 토마토와 똑같은 문제이다.

코드확인

from collections import deque
m,n,h=map(int,input().split())
D=[list(list(map(int,[*map(int,input().split())])) for _ in range(n)) for k in range(h)]
dx,dy,dz=[0,0,0,0,1,-1],[0,0,1,-1,0,0],[-1,1,0,0,0,0]
S=[[[-1]*m for _ in range(n)] for p in range(h)]
q=deque();ans=0;ANS=0
for i in range(h):
    for j in range(n):
        for k in range(m):
            if D[i][j][k]==1:
                q.append((i,j,k))
                S[i][j][k]=0
while q:
    x,y,z=q.popleft()
    for i in range(6):
        nx,ny,nz=x+dx[i],y+dy[i],z+dz[i]
        if 0<=nx<h and 0<=ny<n and 0<=nz<m and S[nx][ny][nz]==-1 and D[nx][ny][nz]==0:
            q.append((nx,ny,nz))
            S[nx][ny][nz]=S[x][y][z]+1
for i in range(h):
    for j in range(n):
        for k in range(m):
            if D[i][j][k]!=-1 and S[i][j][k]==-1:ans=-1
            ANS=max(ANS,S[i][j][k])
if ans==-1:print(-1)
else:print(ANS)