[Python] [BOJ] 13549 숨바꼭질 3

문제

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

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

입력

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

출력

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

---

풀이법

def move_to():
    from heapq import heappush, heappop
    N, K = map(int, input().split())
    # 출발지 >= 도착지일 경우, 순간이동 사용 X
    if N >= K:
        return N - K
    limit = K - N
    time_list = [limit] * (K * 2)
    time_list[K] = 0
    move = []
    heappush(move, (0, K))  # 도착지에서 출발지로 가는 경우 탐색
    while move:
        time, x = heappop(move)
        if x % 2 == 0:  # 현재 2의 배수일 경우, 비교
            half = x // 2
            if time_list[half] > time:
                time_list[half] = time
                heappush(move, (time, half))

        new_time = time + 1
        for nx in (x - 1, x + 1):
            if 0 <= nx < 2 * K and time_list[nx] > new_time:
                time_list[nx] = new_time
                heappush(move, (new_time, nx))
    return time_list[N]


print(move_to())