[Swift] 백준 7569 토마토

import Foundation

let MNH = readLine()!.split(separator: " ").map { Int($0)! }

let M = MNH[0]   // 열
let N = MNH[1]   // 행
let H = MNH[2]   // 면

var board = [[[Int]]]()

for _ in 0..<H {
    var tmp = [[Int]]()
    for _ in 0..<N {
        let line = readLine()!.split(separator: " ").map{ Int($0)! }
        tmp.append(line)
    }
    board.append(tmp)
}

let dx = [-1, 1, 0, 0, 0, 0]
let dy = [0, 0, -1, 1, 0, 0]
let dz = [0, 0, 0, 0, -1, 1]

var queue = [(Int, Int, Int)]()
var dist = Array(repeating: Array(repeating: Array(repeating: -1, count: M), count: N), count: H)
var lastIdx = (0, 0, 0)
var idx = 0
var isCooked = true

for i in 0..<H {
    for j in 0..<N {
        for k in 0..<M {
            if board[i][j][k] == 1 {
                queue.append((i, j, k))
            }
        }
    }
}

while idx<queue.count {

    let cur = queue[idx]
    idx += 1

    for dir in 0..<6 {
        let nz = cur.0 + dz[dir]
        let ny = cur.1 + dy[dir]
        let nx = cur.2 + dx[dir]

        if nx<0 || ny<0 || nz<0 || nx>=M || ny>=N || nz>=H { continue }
        if board[nz][ny][nx] == 0 {
            board[nz][ny][nx] = 1
            queue.append((nz, ny, nx))
            // board[H][N][M] 순서
            dist[nz][ny][nx] = dist[cur.0][cur.1][cur.2] + 1
            lastIdx = (nz, ny, nx)
        }
    }
}

for i in 0..<H {
    for j in 0..<N {
        for k in 0..<M {
            if board[i][j][k] == 0 {
                isCooked = false
            }
        }
    }
}

if isCooked {
    print(dist[lastIdx.0][lastIdx.1][lastIdx.2]+1)
} else {
    print(-1)
}

[BFS] 시작점이 여러개인 유형 + 3차원배열

- board : 3차원배열을 담을 공간
- dx, dy, dz : 방향을 정해준다. (익은 토마토를 결정할 방향을 찾기 위해)
- dist : 토마토가 익은 시간을 담을 배열을 선언해준다. (최소거리와 비슷한 개념)
- lastIdx : 마지막 좌표를 담을 튜플
- idx: 큐의 포인터 인덱스
- isCooked : 모든 토마토가 익었는지 확인하기 위한 Bool

1. 3차원 배열을 돌며 익은토마토(1)가 있는 좌표를 큐에 넣어준다.
2. 큐에서 idx를 이용해 첫번째 요소를 pop 해준다.
3. pop한 좌표를 6가지 방향으로 bfs를 돌린다. (이때 index 순서는 z,y,x 순이다. 처음에 그렇게 담아서..)
4. 새로운 좌표의 값이 안익은토마토(0)일 경우 board의 값을 익은토마토(1)로 바꾸고, dist에 익은 시간을 저장해준다.
lastIdx에는 새로운 좌표를 저장해둔다.
5. board를 돌면서 안익은토마토(0)가 있는 지 확인하고 isCooked에 결과를 저장해둔다.
6. lastIdx를 이용해 출력한다.