[백준] 상범 빌딩 (6593)(Kotlin)

원본 문제

문제 설명

당신은 상범 빌딩에 갇히고 말았다. 여기서 탈출하는 가장 빠른 길은 무엇일까? 상범 빌딩은 각 변의 길이가 1인 정육면체(단위 정육면체)로 이루어져있다. 각 정육면체는 금으로 이루어져 있어 지나갈 수 없거나, 비어있어서 지나갈 수 있게 되어있다. 당신은 각 칸에서 인접한 6개의 칸(동,서,남,북,상,하)으로 1분의 시간을 들여 이동할 수 있다. 즉, 대각선으로 이동하는 것은 불가능하다. 그리고 상범 빌딩의 바깥면도 모두 금으로 막혀있어 출구를 통해서만 탈출할 수 있다.

당신은 상범 빌딩을 탈출할 수 있을까? 만약 그렇다면 얼마나 걸릴까?

입력

입력은 여러 개의 테스트 케이스로 이루어지며, 각 테스트 케이스는 세 개의 정수 L, R, C로 시작한다. L(1 ≤ L ≤ 30)은 상범 빌딩의 층 수이다. R(1 ≤ R ≤ 30)과 C(1 ≤ C ≤ 30)는 상범 빌딩의 한 층의 행과 열의 개수를 나타낸다.

그 다음 각 줄이 C개의 문자로 이루어진 R개의 행이 L번 주어진다. 각 문자는 상범 빌딩의 한 칸을 나타낸다. 금으로 막혀있어 지나갈 수 없는 칸은 ‘#’으로 표현되고, 비어있는 칸은 ‘.’으로 표현된다. 당신의 시작 지점은 ‘S’로 표현되고, 탈출할 수 있는 출구는 ‘E’로 표현된다. 각 층 사이에는 빈 줄이 있으며, 입력의 끝은 L, R, C가 모두 0으로 표현된다. 시작 지점과 출구는 항상 하나만 있다.

출력

각 빌딩에 대해 한 줄씩 답을 출력한다. 만약 당신이 탈출할 수 있다면 다음과 같이 출력한다.

Escaped in x minute(s).

여기서 x는 상범 빌딩을 탈출하는 데에 필요한 최단 시간이다. 만일 탈출이 불가능하다면, 다음과 같이 출력한다.

Trapped!

테스트 케이스1

입력	출력
3 4 5 S…. .###. .##.. ###.# ##### ##### ##.## ##… ##### ##### #.### ####E 1 3 3 S## #E# ### 0 0 0	Escaped in 11 minute(s). Trapped!

문제 풀이1 (BFS)

import java.io.BufferedReader
import java.io.InputStreamReader
import java.util.*

var building: Array<Array<CharArray>> = arrayOf()
var route: Array<Array<IntArray>> = arrayOf()

val dx = intArrayOf(1,-1,0,0,0,0)
val dy = intArrayOf(0,0,1,-1,0,0)
val dz = intArrayOf(0,0,0,0,1,-1)

var end: Node = Node(0, 0, 0)
var start: Node = Node(0, 0, 0)

data class Node(val x: Int, val y: Int, val z: Int)

fun main() = with(BufferedReader(InputStreamReader(System.`in`))) {

    while (true) {
        var x = 0
        var y = 0
        var z = 0

        var intArr = readLine()
        if (intArr.isEmpty()) {
            val getValue = readLine()
                .split(" ")
                .map { it.toInt() }
            z = getValue[0]
            y = getValue[1]
            x = getValue[2]
        } else {
            val getValue = intArr
                .split(" ")
                .map { it.toInt() }
            z = getValue[0]
            y = getValue[1]
            x = getValue[2]
        }

        if (z == 0 && y == 0 && x == 0) break

        building = Array(z) { Array(y) { charArrayOf() } }
        route = Array(z) { Array(y) { IntArray(x) { Int.MAX_VALUE } } }

        for (dz in 0 until z) {
            for (dy in 0 until y) {
                var charArr = readLine()
                while (charArr.isEmpty()) charArr = readLine()
                building[dz][dy] = charArr
                    .toCharArray()
                if (charArr.contains('S')) {
                    for (dx in 0 until x) {
                        if (charArr[dx] == 'S')
                            start = Node(dx, dy, dz)
                    }
                }
                if (charArr.contains('E')) {
                    for (dx in 0 until x) {
                        if (charArr[dx] == 'E')
                            end = Node(dx, dy, dz)
                    }
                }
            }
        }
        bfs(x, y, z)
    }
}

fun bfs(x: Int, y: Int, z: Int) {
    val queue: LinkedList<Node> = LinkedList()
    queue.add(start)
    route[start.z][start.y][start.x] = 0

    while (queue.isNotEmpty()) {
        val (current_x, current_y, current_z) = queue.poll()

        for (i in 0 until 6) {
            val nx = current_x + dx[i]
            val ny = current_y + dy[i]
            val nz = current_z + dz[i]

            if (nx !in 0 until x || ny !in 0 until y || nz !in 0 until z) continue
            if (building[nz][ny][nx] == '#') continue

            if (route[nz][ny][nx] > route[current_z][current_y][current_x] + 1) {
                route[nz][ny][nx] = route[current_z][current_y][current_x] + 1
                queue.add(Node(nx, ny, nz))
            }
        }
    }
    if (route[end.z][end.y][end.x] == Int.MAX_VALUE)
        println("Trapped!")
    else
        println("Escaped in ${route[end.z][end.y][end.x]} minute(s).")
}