Koala - 15기/코딩테스트 준비 스터디

[BOJ/Python] 13849번: 숨바꼭질 3

haerr 2024. 8. 18. 20:41

https://www.acmicpc.net/problem/13549

문제

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

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

 

입력

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

 

출력

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

 

소스코드

import heapq
INF = int(1e9)

N, K = map(int, input().split()) 
distance = [INF]*100001  

def dijkstra(start): 
    distance[start] = 0  # 시작 위치 초기화
    q = []
    heapq.heappush(q, (0, start))  

    while q: 
        dist, now = heapq.heappop(q)  
        if distance[now] < dist:
            continue
        for a, b in [(now*2, dist), (now+1, dist+1), (now-1, dist+1)]:  
            if 0 <= a <= 100000 and distance[a] > b:  
                distance[a] = b
                heapq.heappush(q, (b, a))

dijkstra(N)  
print(distance[K])

 

풀이

1. 우선순위 큐를 구현하기 위해 heapq import

2. dijkstra 함수에서 다익스트라 알고리즘 구현

3. 시작 위치와 그 거리를 우선순위 큐에 추가하고 q가 존재할때, 현재위치에서 이동가능한 3가지 경우에서 다음범위가 유효하고 더 짧다면 값을 업데이트함.

4. 출력