[Python/파이썬] 백준 13549번 숨바꼭질 3

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

출력

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

코드

• 0-1 BFS

from collections import deque

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

q = deque([[n, 0]])
visited = [False] * 100_001
visited[n] = True
while q:
    now, cnt = q.popleft()
    if now == k:
        print(cnt)
        break
    if 0 <= now * 2 <= 100_000 and not visited[now * 2]:
        q.appendleft([now * 2, cnt])
        visited[now * 2] = True
    for i in [now - 1, now + 1]:
        if 0 <= i <= 100_000 and not visited[i]:
            q.append([i, cnt + 1])
            visited[i] = True

 

• 다익스트라

import heapq

n, k = map(int, input().split())
visited = [False] * 100_001

q = [(0, n)]
while True:
    cnt, now = heapq.heappop(q)
    if now == k:
        print(cnt)
        break
    visited[now] = True
    if now * 2 <= 100_000 and not visited[now*2]:
        heapq.heappush(q, (cnt, now*2))
    for i in [now-1, now+1]:
        if 0 <= i <= 100_000 and not visited[i]:
            heapq.heappush(q, (cnt+1, i))