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. 출력