백준 7569 토마토2

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

7569번: 토마토

전의 토마토와 완전히 똑같은 방식이다.

다만, 이번에는 z축이 추가됬다.

i,j,k로 인덱싱하고,

움직이는방향을 [[-1,0,0],[0,-1,0],[0,0,-1],[1,0,0],[0,1,0],[0,0,1]]

6방향으로 bfs를 진행하면 된다.

진행할때마다 얼마나 날짜가 지났는지까지 세주면 완벽

 

마지막에, 하나씩탐색하다가 0이 있으면 -1출력하고 종료

import sys
input=sys.stdin.readline
from collections import deque
moves=[[-1,0,0],[0,-1,0],[0,0,-1],[1,0,0],[0,1,0],[0,0,1]]

def bfs(q):
    day=0
    while q:
        i,j,k,v=q.popleft()
        for move in moves:
            a,b,c=move
            ti=a+i
            tj=b+j
            tk=c+k
            if 0<=ti<h and 0<=tj<n and 0<=tk<m and arr[ti][tj][tk]=='0':
                q.append([ti,tj,tk,v+1])
                arr[ti][tj][tk]='-1'
              
    for i in range(h):
        for j in range(n):
            for k in range(m):
                if arr[i][j][k]=='0':
                    print(-1)
                    return
            
    print(v)
    



    



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

arr=[[list(input().strip().split())for _ in range(n)]for _ in range(h)]

q=deque()
answer=0

for i in range(h):
    for j in range(n):
        for k in range(m):
            if arr[i][j][k]=='1':
                q.append((i,j,k,0))
                arr[i][j][k]='-1'
            
bfs(q)