2589번: 보물섬 - Kotlin (BFS)

2589번: 보물섬

문제

보물섬 지도를 발견한 후크 선장은 보물을 찾아나섰다. 보물섬 지도는 아래 그림과 같이 직사각형 모양이며 여러 칸으로 나뉘어져 있다. 각 칸은 육지(L)나 바다(W)로 표시되어 있다. 이 지도에서 이동은 상하좌우로 이웃한 육지로만 가능하며, 한 칸 이동하는데 한 시간이 걸린다. 보물은 서로 간에 최단 거리로 이동하는데 있어 가장 긴 시간이 걸리는 육지 두 곳에 나뉘어 묻혀있다. 육지를 나타내는 두 곳 사이를 최단 거리로 이동하려면 같은 곳을 두 번 이상 지나가거나, 멀리 돌아가서는 안 된다.

예를 들어 위와 같이 지도가 주어졌다면 보물은 아래 표시된 두 곳에 묻혀 있게 되고, 이 둘 사이의 최단 거리로 이동하는 시간은 8시간이 된다.

보물 지도가 주어질 때, 보물이 묻혀 있는 두 곳 간의 최단 거리로 이동하는 시간을 구하는 프로그램을 작성하시오.

입력

첫째 줄에는 보물 지도의 세로의 크기와 가로의 크기가 빈칸을 사이에 두고 주어진다. 이어 L과 W로 표시된 보물 지도가 아래의 예와 같이 주어지며, 각 문자 사이에는 빈 칸이 없다. 보물 지도의 가로, 세로의 크기는 각각 50이하이다.

 

나의 풀이

요즘 나는 브루트 포스 문제 알고리즘을 해결하려고 브루트 포스 문제를 풀고 있었다. 그런데, 이 문제는 그래프 탐색 이론을 이용해야 하는 문제였다. 문제를 읽어보니 최단 거리를 구해야 함으로 BFS를 이용하고 그냥 BFS를 사용했을 때는 모든 장소를 탐색할 수 없음으로 브루트 포스 알고리즘으로 모든 장소에 대한 경우의 수를 구하여 배열에 넣어주고 최대값을 구해 주었다.

 

import java.util.*
import kotlin.collections.ArrayList

val dx = arrayOf(0, 0, 1, -1)
val dy = arrayOf(1, -1, 0, 0)
val answer = ArrayList<Int>()

fun main() = with(System.`in`.bufferedReader()) {
    val (n, m) = readLine().split(" ").map { it.toInt() }

    val map = Array(n) { readLine().toCharArray() }

    for (i in 0 until n) {
        for (j in 0 until m) {
            if (map[i][j] == 'L') {
                bfs(i, j, n, m, map)
            }
        }
    }
    print(answer.max())
}

fun bfs(x: Int, y: Int, n: Int, m: Int, map: Array<CharArray>) {
    val queue: Queue<Triple<Int, Int, Int>> = LinkedList()
    val visited = Array(n) { Array(m) { false } }
    queue.add(Triple(x, y, 0))
    visited[x][y] = true

    var count = 0
    while (queue.isNotEmpty()){
        val now = queue.poll()
        count = now.third

        for(i in 0 until 4){
            val tempX = now.first + dx[i]
            val tempY = now.second + dy[i]

            if(tempX in 0 until n
                && tempY in 0 until m
                && !visited[tempX][tempY]
                && map[tempX][tempY] == 'L' ){
                visited[tempX][tempY] = true
                queue.add(Triple(tempX, tempY, now.third + 1))
            }
        }
    }
    answer.add(count)
}