[Baekjoon] 13913번 숨바꼭질 3

문제

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

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

입력

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

출력

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

예제

Example 1:

Input: 
5 17
Output: 
2

조건

시간 제한 : 2초

메모리 제한 : 512 MB

풀이과정

간선의 가중치가 같다면 BFS를 사용할 수 있지만 0과 1로 다르기 때문에 다른 방식을 적용하여야 한다.

내 풀이1 (다익스트라/우선순위 큐)

import heapq

# 무한대 정의
INF = int(1e9)

# n = 수빈이 현재 위치, k = 동생 현재 위치
n, k = map(int,input().split())

# 시간 정보에 대한 테이블
distance = [INF for _ in range(100001)]

q = []
heapq.heappush(q,(0, n))
distance[n] = 0
while q:
    dist, now = heapq.heappop(q)
    for i in [2*now, now+1, now-1]:
        if 0 <= i <= 100000 and distance[i] == INF:
            if i == 2*now:
                distance[i] = dist
                heapq.heappush(q, (dist, i))
            else:
                distance[i] = dist + 1
                heapq.heappush(q, (dist+1, i))

print(distance[k])

내 풀이2 (0-1 BFS)

from collections import deque

# 무한대 정의
INF = int(1e9)

# n = 수빈이 현재 위치, k = 동생 현재 위치
n, k = map(int,input().split())

# 시간 정보에 대한 테이블
distance = [INF for _ in range(100001)]


# 0 - 1 BFS
q = deque()
q.append((0, n))
distance[n] = 0
while q:
    dist, now = q.popleft()
    for i in [2*now, now+1, now-1]:
        if 0 <= i <= 100000 and distance[i] == INF:
            if i == 2*now:
                distance[i] = dist
                q.appendleft((dist, i))
            else:
                distance[i] = dist + 1
                q.append((dist + 1, i))

print(distance[k])