[백준] 최단거리 13549 숨바꼭질 3 파이썬 풀이

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

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는 정수이다.

출력

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

예제 입력 1 복사

5 17

예제 출력 1 복사

2

 

일단 이게 처음 푼 내 코드이다.

import sys
# sys.stdin.readline()
from collections import deque
sys.setrecursionlimit(100000000)
import heapq
INF = int(1e9)
N,K = map(int,sys.stdin.readline().split())
graph = [[] for _ in range(100001)]
graph[0].append((1,1))
graph[100000].append((99999,1))
for i in range(1,100000):
    if i <= 50000:
     graph[i].append((i-1,1))
     graph[i].append((i+1,1))
     graph[i].append((2*i,0))
    else:
        graph[i].append((i-1,1))
        graph[i].append((i+1,1))


distance = [INF] * 100001

def djikstra(start):
    q = []
    heapq.heappush(q,(0,start))
    distance[start] = 0
    while q:
        dist, now = heapq.heappop(q)
        if distance[now] < dist:
            continue
        for i in graph[now]:
            cost = dist + i[1]
            if cost < distance[i[0]]:
                distance[i[0]] = cost
                heapq.heappush(q,(cost,i[0]))

djikstra(N)
print(distance[K])

반복문을 돌리는 과정에서 i가 5만을 넘어가면 , 2*i로 도약할 수 없어서 50000을 기점으로 케이스를 나눠줬다.(N이 100000 이하의 숫자이므로)

반복문에서 제외되는 0과 100000은 따로 추가

그 뒤로는 그냥 다익스트라 알고리즘을 사용했다 .

출발점을 N으로 해서 K까지의 최단 거리를 구하면 되는 쉬운 문제.

가중치만 0과 1로 잘 설정해주면 되는 것 같다.