[백준] 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는 정수이다.

출력

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

풀이

# [백준] 13549번 숨바꼭질3
import heapq

INF = int(1e9)

MAX = 1000000

graph = [INF]* (MAX +1)

n, k = map(int,input().split())

def solve(start):
    q = []
    heapq.heappush(q,(0,start))
    while q:
        cnt, now = heapq.heappop(q)
        if now == k:
            return cnt
        for i in (now-1, now +1, now*2):
            if 0<= i <= MAX:
                if graph[i] > cnt:
                    graph[i] = cnt
                    if i == now*2:
                        heapq.heappush(q,(cnt, i))
                    else:
                        heapq.heappush(q,(cnt+1, i))
print(solve(n))

후기

• 해당 문제를 풀기 위해서 고민을 했었는데, 예전에 숨바꼭질 1 를 풀었던 것을 참고해서 문제를 새롭게 풀어보았다. 그렇게 했더니 비교적 쉽게 풀려서 다행이긴 한데, 참고를 안하고 풀고 싶었는데, 그 부분이 조금 아쉽다.