[백준][코틀린] 1697 숨바꼭질

문제 내용

문제

수빈이는 동생과 숨바꼭질을 하고 있다. 수빈이는 현재 점 N(0 ≤ N ≤ 100,000)에 있고, 동생은 점 K(0 ≤ K ≤ 100,000)에 있다. 수빈이는 걷거나 순간이동을 할 수 있다. 만약, 수빈이의 위치가 X일 때 걷는다면 1초 후에 X-1 또는 X+1로 이동하게 된다. 순간이동을 하는 경우에는 1초 후에 2*X의 위치로 이동하게 된다.

수빈이와 동생의 위치가 주어졌을 때, 수빈이가 동생을 찾을 수 있는 가장 빠른 시간이 몇 초 후인지 구하는 프로그램을 작성하시오.

입력

첫 번째 줄에 수빈이가 있는 위치 N과 동생이 있는 위치 K가 주어진다. N과 K는 정수이다.

출력

수빈이가 동생을 찾는 가장 빠른 시간을 출력한다.

예제 입력	예제 출력
5 17	4

 

문제 풀이

BFS로 최단 거리를 구하였다. 큐에 다음으로 가는 좌표를 저장하는데, 이때 이동 시간도 함께 저장을 해준다. 

큐에서 하나씩 poll 할 때 동생이 있는 곳이라면, 시간을 출력해준다. 

동생이 있는 곳이 아니라면 다음 좌표가 범위내이고, 간적이 없는 곳이라면 다음 좌표를 큐에 저장시킨다. 

</>̆̈ 코드

import java.util.*

fun main() {
    print(bfs())
}

private fun bfs(): Int {
    val (n, k) = readLine()!!.split(" ").map { it.toInt() }
    val visited = BooleanArray(100_001)
    val queue = LinkedList<Pair<Int, Int>>()
    queue.add(Pair(n, 0)); visited[n] = true

    while (queue.isNotEmpty()) {
        val (newPoint, time) = queue.poll()
        if (newPoint == k) return time

        if (checkRange(newPoint+1) && !visited[newPoint+1]) {
            queue.add(Pair(newPoint+1, time+1)); visited[newPoint+1] = true
        }
        if (checkRange(newPoint-1) && !visited[newPoint-1]) {
            queue.add(Pair(newPoint-1, time+1)); visited[newPoint-1] = true
        }
        if (checkRange(newPoint*2) && !visited[newPoint*2]) {
            queue.add(Pair(newPoint*2, time+1)); visited[newPoint*2] = true
        }
    }

    return -1
}

private fun checkRange(x: Int) = x in 0 .. 100_000

링크

1697번: 숨바꼭질