백준 2589 [kotlin]

문제

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

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

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

입력

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

출력

첫째 줄에 보물이 묻혀 있는 두 곳 사이를 최단 거리로 이동하는 시간을 출력한다.

코드

import java.lang.Integer.max
import java.util.LinkedList
import java.util.Queue

var dx = intArrayOf(-1,1,0,0)
var dy = intArrayOf(0,0,1,-1)
fun main()
{
    var xy = readLine()!!.split(" ").map {it.toInt()}
    var x = xy[0]
    var y = xy[1]

    var answer = 0
    var tresure : Array<CharArray> = Array(x, { CharArray(y,{'W'}) })
    repeat(x)
    {
        i ->
        var k = readLine()!!
        repeat(y)
        { j ->
            tresure[i][j] = k[j]
        }
    }

    repeat(x)
    {
            i ->
        repeat(y){
                j ->
            if (tresure[i][j] == 'L')
            {
                answer = max(bfs(tresure,x,y,i,j), answer)
            }
        }
    }
    println(answer)
}

fun bfs(trs:Array<CharArray>, M:Int, N:Int,x:Int, y:Int) : Int {
    var check : Array<IntArray> = Array(M, { IntArray(N,{2500}) } )
    var Q:Queue<IntArray> = LinkedList()
    var ans = 0
    check[x][y] = 1
    Q.add(intArrayOf(x,y))
    while(Q.isNotEmpty())
    {
        var k = Q.poll()
        var qx = k[0]
        var qy = k[1]
        for(i in 0..3)
        {
            var nx = qx + dx[i]
            var ny = qy + dy[i]
            if (nx in 0..M-1 && ny in 0..N-1 && trs[nx][ny] == 'L' && check[qx][qy] + 1 < check[nx][ny])
            {
                ans = max(ans, check[qx][qy] + 1)
                check[nx][ny] = check[qx][qy] + 1
                Q.add(intArrayOf(nx,ny))
            }
        }
    }
    return ans - 1
}

삽질

bfs를 구현하면서 이동 거리를 기록해둘 배열을 생성할때 이동거리는 가로 세로는 각각 50이라는 말에 단순하게 최대 100으로 100으로 제한을 해두면 거기까지는 최대 이동거리가 아니겠지 라고 생각한게 바보같은 생각이었다. 체점을 돌리자마자 5%에서 바로 틀려버렸다. 좀 넉넉하게 2500으로 두니까 바로 성공했다.