[백준][코틀린] 12851 숨바꼭질 2

문제 내용

문제

수빈이는 동생과 숨바꼭질을 하고 있다. 수빈이는 현재 점 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 2

문제 풀이

2022.02.02 - [Algorithm/with Kotlin] - [백준][코틀린] 1697 숨바꼭질

위 문제의 연관 문제이다! 위 문제와 다른 점은 가장 빠른 시간으로 수빈이가 동생을 찾는 방법의 수를 출력해야된다는 것이다. 

예를 들어서 수빈이가 있는 위치가 1이고, 동생이 있는 위치가 4일 때 갈수 있는 경우의 수는 다음 두가지 이다.

1. 1 → (1 + 1 = ) 2 → (2 * 2 = ) 4
2. 1 → (1 * 2 = ) 2 → (2 * 2 = ) 4

아래 코드 (숨바꼭질 1 코드중 일부)처럼 queue에 push 할때마다 방문 지점을 표시하면 첫번째 경우만 고려하고 두번째 경우는 체킹을 하지 못한다. 

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

따라서, 큐에서 값을 뺄 때 방문지점을 true로 변경해주어야, 첫번째 두번째 경우 모두 고려할 수 있다! 

</>̆̈ 코드

import java.util.*

fun main() {
    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

    var minimum = Int.MAX_VALUE; var cnt = 0

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

        visited[newPoint] = true

        if (newPoint == k) {
            if (minimum > time) {
                minimum = time; cnt = 1
            } else if (minimum == time) cnt ++
            continue
        }

        if (checkRange(newPoint+1) && !visited[newPoint+1]) queue.add(Pair(newPoint+1, time+1))

        if (checkRange(newPoint-1) && !visited[newPoint-1]) queue.add(Pair(newPoint-1, time+1))

        if (checkRange(newPoint*2) && !visited[newPoint*2])queue.add(Pair(newPoint*2, time+1))
    }
    print("$minimum\n$cnt")
}

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

링크

12851번: 숨바꼭질 2