백준 6593 상범 빌딩 Kotlin (bfs)

문제 출처 : https://www.acmicpc.net/problem/6593

6593번: 상범 빌딩

문제

당신은 상범 빌딩에 갇히고 말았다. 여기서 탈출하는 가장 빠른 길은 무엇일까? 상범 빌딩은 각 변의 길이가 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!

알고리즘 분류

• 그래프 이론
• 그래프 탐색
• 너비 우선 탐색

풀이

간단한 bfs문제이다.

3차원이라 좀 재밌었다.

출력 형식에 주의하자. 출력 형식 땜에 맞왜틀 계속 외쳤는데 minutes가 아니라 minute였다.

첨엔 . 안 붙여서 틀렸었다.

음.. 딱히 설명할 게 없다 그냥  bfs 기본 문제이다.

bfs에 대해 안다면 가볍게 풀 수 있기 때문에 위에 있는 사항만 주의하면 된다.

본인은 큐에 moveCnt를 넣지 않고 q.size를 이용하는 테크닉으로 최소 이동 거리를 구했다.

코드

import java.util.*
val br = System.`in`.bufferedReader()
fun getIntList() = br.readLine().split(' ').map { it.toInt() }
fun getInt() = br.readLine().toInt()

lateinit var graph: Array<Array<CharArray>>
var sl = 0
var sr = 0
var sc = 0
var el = 0
var er = 0
var ec = 0
val dir = arrayOf(
    arrayOf(0,1),
    arrayOf(0,-1),
    arrayOf(1,0),
    arrayOf(-1,0)
)
fun bfs(h: Int, n: Int, m: Int) : Int{
    var moveCnt = 0
    val q: Queue<Triple<Int,Int,Int>> = LinkedList()
    q.add(Triple(sl,sr,sc))
    graph[sl][sr][sc] = '#'
    while(q.isNotEmpty()){
        val size = q.size
        for(i in 0 until size) {
            val (l, r, c) = q.poll()
            for (j in 0 until 4) {
                val nr = r + dir[j][0]
                val nc = c + dir[j][1]
                if (nr !in 0 until n || nc !in 0 until m) continue
                if (graph[l][nr][nc] == '#') {
                    continue
                } else if (graph[l][nr][nc] == 'E') {
                    return moveCnt + 1
                }
                graph[l][nr][nc] = '#'
                q.add(Triple(l, nr, nc))
            }
            var nl = l + 1
            if (nl in 0 until h) {
                if (graph[nl][r][c] == 'E') {
                    return moveCnt + 1
                } else if (graph[nl][r][c] == '.') {
                    graph[nl][r][c] = '#'
                    q.add(Triple(nl, r, c))
                }
            }
            nl = l - 1
            if (nl in 0 until h) {
                if (graph[nl][r][c] == 'E') {
                    return moveCnt + 1
                } else if (graph[nl][r][c] == '.') {
                    graph[nl][r][c] = '#'
                    q.add(Triple(nl, r, c))
                }
            }
        }
        moveCnt++
    }
    return -1
}

fun main() = with(System.out.bufferedWriter()){
    while(true) {
        //input
        val (l, r, c) = getIntList()
        if (l == 0 && r == 0 && c == 0) break
        graph = Array(l) { i ->
            val floor = Array(r) { j->
                val line = br.readLine()
                for(k in line.indices){
                    if(line[k] == 'S'){
                        sl = i
                        sr = j
                        sc = k
                    }
                    else if(line[i] == 'E'){
                        el = i
                        er = j
                        ec = k
                    }
                }
                line.toCharArray()
            }
            br.readLine()
            floor
        }
        //solve
        val answer = bfs(l,r,c)

        //output
        if(answer>=0){
            write("Escaped in $answer minute(s).\n")
        }else{
            write("Trapped!\n")
        }
    }
    close()
}