[백준 7569/파이썬] 토마토

소스코드

import sys

from collections import deque

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

graph = []

queue = deque([])

 

for i in range(h):

    tmp = []

    for j in range(n):

        tmp.append(list(map(int,sys.stdin.readline().split())))

        for k in range(m):

            if tmp[j][k]==1:

                queue.append([i,j,k])

    graph.append(tmp)

    

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

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

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

while(queue):

    x,y,z = queue.popleft()

    

    for i in range(6):

        a = x+dx[i]

        b = y+dy[i]

        c = z+dz[i]

        if 0<=a<h and 0<=b<n and 0<=c<m and graph[a][b][c]==0:

            queue.append([a,b,c])

            graph[a][b][c] = graph[x][y][z]+1

            

day = 0

for i in graph:

    for j in i:

        for k in j:

            if k==0:

                print(-1)

                exit(0)

        day = max(day,max(j))

print(day-1)

알고리즘

1. BFS를 사용하기 위해서 queue를 이용한다.

2. 입력받을때 sys.stdin.readline()을 사용하여 시간을 줄여준다. 또 입력이 1 이면 큐에 저장한다.

3. 큐에 들어온 아이 기준 상하좌우앞뒤가 0이면 또 큐에 넣는다.

4. 그러면 while 문 종료 조건이 queue가 공백일때가 되니 다 상할때까지 반복문이 실행이 되겠죵?