미로탈출 - Kotlin (BFS)

프로그래머스

문제 설명

1 x 1 크기의 칸들로 이루어진 직사각형 격자 형태의 미로에서 탈출하려고 합니다. 각 칸은 통로 또는 벽으로 구성되어 있으며, 벽으로 된 칸은 지나갈 수 없고 통로로 된 칸으로만 이동할 수 있습니다. 통로들 중 한 칸에는 미로를 빠져나가는 문이 있는데, 이 문은 레버를 당겨서만 열 수 있습니다. 레버 또한 통로들 중 한 칸에 있습니다. 따라서, 출발 지점에서 먼저 레버가 있는 칸으로 이동하여 레버를 당긴 후 미로를 빠져나가는 문이 있는 칸으로 이동하면 됩니다. 이때 아직 레버를 당기지 않았더라도 출구가 있는 칸을 지나갈 수 있습니다. 미로에서 한 칸을 이동하는데 1초가 걸린다고 할 때, 최대한 빠르게 미로를 빠져나가는데 걸리는 시간을 구하려 합니다.

미로를 나타낸 문자열 배열 maps가 매개변수로 주어질 때, 미로를 탈출하는데 필요한 최소 시간을 return 하는 solution 함수를 완성해주세요. 만약, 탈출할 수 없다면 -1을 return 해주세요.

제한사항

• 5 ≤ maps의 길이 ≤ 100
  • 5 ≤ maps[i]의 길이 ≤ 100
  • maps[i]는 다음 5개의 문자들로만 이루어져 있습니다.
    • S : 시작 지점
    • E : 출구
    • L : 레버
    • O : 통로
    • X : 벽
  • 시작 지점과 출구, 레버는 항상 다른 곳에 존재하며 한 개씩만 존재합니다.
  • 출구는 레버가 당겨지지 않아도 지나갈 수 있으며, 모든 통로, 출구, 레버, 시작점은 여러 번 지나갈 수 있습니다.

 

 

나의 풀이

그래프 탐색을 할 줄 알면 쉽게 해결할 수 있는 문제이다. 다만 최단거리를 구해야 함으로 BFS를 사용하는 것이 좋다. 이 문제는 출발지에서 레버까지, 그리고 레버에서 도착지까지의 거리 2개를 구한 다음, 그 거리를 더하면 되는 문제이다. 

 

BFS를 재사용할 수 있게 만들면 더 좋다.

 

import java.util.*

class Solution {
    val dx = arrayOf(0, 0, 1, -1)
    val dy = arrayOf(1, -1, 0, 0)

    fun solution(maps: Array<String>): Int {
        var start = Pair(0, 0)
        var exit = Pair(0, 0)
        var lever = Pair(0, 0)

        for (i in maps.indices) {
            for (j in maps[0].indices) {
                if (maps[i][j] == 'S')
                    start = Pair(i, j)
                else if (maps[i][j] == 'E')
                    exit = Pair(i, j)
                else if (maps[i][j] == 'L')
                    lever = Pair(i, j)
            }
        }

        val timeToLever = bfs(start, lever, maps)
        val timeToExit = bfs(lever, exit, maps)

        return if (timeToLever == -1 || timeToExit == -1)
            -1
        else
            timeToLever + timeToExit
    }


    fun bfs(start: Pair<Int, Int>, end: Pair<Int, Int>, map: Array<String>): Int {
        val queue : Queue<Pair<Int, Int>> = LinkedList()
        val visited = Array(map.size) { IntArray(map[0].length) }

        queue.add(start)
        visited[start.first][start.second] = 1

        while (queue.isNotEmpty()) {
            val (x, y) = queue.poll()

            if (x == end.first && y == end.second)
                return visited[end.first][end.second] - 1


            for(i in 0 until 4){
                val tempX = x + dx[i]
                val tempY = y + dy[i]

                if(tempX in map.indices
                    && tempY in map[0].indices
                    && visited[tempX][tempY] == 0
                    && map[tempX][tempY] != 'X' ){
                    visited[tempX][tempY] = visited[x][y] + 1
                    queue.add(tempX to tempY)
                }
            }
        }

        return -1
    }
}