(Python) - BOJ(7569번) : 토마토 II

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

7569번: 토마토

---

• 해설 : 

철수의 토마토 농장에서는 토마토를 보관하는 큰 창고를 가지고 있다. 토마토는 아래의 그림과 같이 격자모양 상자의 칸에 하나씩 넣은 다음, 상자들을 수직으로 쌓아 올려서 창고에 보관한다.

창고에 보관되는 토마토들 중에는 잘 익은 것도 있지만, 아직 익지 않은 토마토들도 있을 수 있다. 보관 후 하루가 지나면, 익은 토마토들의 인접한 곳에 있는 익지 않은 토마토들은 익은 토마토의 영향을 받아 익게 된다. 하나의 토마토에 인접한 곳은 위, 아래, 왼쪽, 오른쪽, 앞, 뒤 여섯 방향에 있는 토마토를 의미한다. 대각선 방향에 있는 토마토들에게는 영향을 주지 못하며, 토마토가 혼자 저절로 익는 경우는 없다고 가정한다. 철수는 창고에 보관된 토마토들이 며칠이 지나면 다 익게 되는지 그 최소 일수를 알고 싶어 한다.

토마토를 창고에 보관하는 격자모양의 상자들의 크기와 익은 토마토들과 익지 않은 토마토들의 정보가 주어졌을 때, 며칠이 지나면 토마토들이 모두 익는지, 그 최소 일수를 구하는 프로그램을 작성하라. 단, 상자의 일부 칸에는 토마토가 들어있지 않을 수도 있다.

 

 

 

---

• 풀이 :

bfs문제인데 3차배열의 bfs라 조금 까다로운 문제이다. 원래의 bfs가 가로, 세로의 4방향 이동만 고려해주었다면 이 문제는 x축, y축, z축의 3가지 경우인 6방위를 고려해주어야 한다. 

이 개념만 이해한다면 나머지는 쉽게 구현할 수 있는 bfs 기초 문제이다.

 

 

 

---

• 소스코드 : 

 

import sys
from collections import deque
input = sys.stdin.readline
 
dx = [-1, 1, 0, 0, 0, 0]
dy = [0, 0, -1, 1, 0, 0]
dz = [0, 0, 0, 0, -1, 1]
# [층수][세로][가로]
 
if __name__ =="__main__":
    N,M,H = map(int,input().split())
    t = [[list(map(int,input().split())) for _ in range(M)] for _ in range(H)]
    queue = deque([])
    c = [[[0 for _ in range(N)] for _ in range(M)] for _ in range(H)]
    flag = True
    for i in range(H):
        for j in range(M):
            for k in range(N):
                if t[i][j][k] == 1:
                    c[i][j][k] = 1
                    queue.append([i,j,k])
    ans = 1
    while queue:
        z,x,y = queue.popleft()
        for i in range(6):
            az = z + dz[i]
            ax = x + dx[i]
            ay = y + dy[i]
            if 0<=az<H and 0<=ax<M and 0<=ay<N:
                if t[az][ax][ay] == 0 and c[az][ax][ay] == 0:
                    queue.append([az,ax,ay])
                    c[az][ax][ay] = c[z][x][y] + 1
                    t[az][ax][ay] = 1
                    ans = max(ans,c[z][x][y] + 1)
    for i in range(H):
        for j in range(M):
            if 0 in t[i][j]:
                print(-1)
                exit()
    print(ans-1)