boj.kr/7569 토마토 (gold5) 파이썬 풀이

1. 3차원 배열로 토마토 정보 초기화

2. 방문체크용 3차원 배열을 초기화

3. 익은 토마토 좌표를 저장하고 방문체크좌표값에 +1

4. 인접한 6개 토마토를 익힌 결과를 bfs방식으로 저장

5. 안익은 토마토가 있는경우 -1 출력, 없다면 방문체크좌표값에 저장된 최대값-1을 출력 

 

M, N, H = map(int, input().split())
graph = [[list(map(int, input().split())) for _ in range(N)] for _ in range(H)]
visited = [[[0]*(M) for _ in range(N)] for _ in range(H)]

dz = (0, 0, 0, 0, -1, 1)
dy = (-1, 1, 0, 0, 0, 0)
dx = (0, 0, -1, 1, 0, 0)
from collections import deque
def bfs():
    dq = deque()
    for z in range(H):
        for y in range(N):
            for x in range(M):
                if graph[z][y][x] == 1:
                    dq.append((z, y, x))
                    visited[z][y][x] += 1
    while dq:
        (cz, cy, cx) = dq.popleft()
        for i in range(6):
            nz = cz + dz[i]
            ny = cy + dy[i]
            nx = cx + dx[i]
            if 0<=nz<H and 0<=ny<N and 0<=nx<M and graph[nz][ny][nx] == 0 and visited[nz][ny][nx] == 0:
                visited[nz][ny][nx] = visited[cz][cy][cx] + 1
                graph[nz][ny][nx] = 1
                dq.append((nz, ny, nx))
    ans = -1e9
    for z2 in range(H):
        for y2 in range(N):
            for x2 in range(M):
                if graph[z2][y2][x2] == 0:
                    return 0
                else:
                    ans = visited[z2][y2][x2] if visited[z2][y2][x2] > ans else ans
    return ans

print(bfs()-1)

 

3차원 배열은 z->y->z축 순으로 순회할 수 있다.

bfs 에서 방문체크 배열에 이동거리를 카운트할 수 있다.