[BaekJoon] 13549 숨바꼭질 3 (Gold 5) - Python

문제

 

수빈이는 동생과 숨바꼭질을 하고 있다. 수빈이는 현재 점 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

---

코드

 

from collections import deque
import sys

input = sys.stdin.readline

N, K = map(int, input().split())

def bfs():

    q = deque()
    q.append(N)
    visited = [-1 for _ in range(100001)]
    visited[N] = 0

    while q:
        x = q.popleft()
        if x == K: 
            return visited[x]
        for i in [x * 2, x + 1, x - 1]:
            if 0 <= i < 100001 and visited[i] == -1:
                if i == x * 2:
                    visited[i] = visited[x]
                    q.appendleft(i)
                else:
                    visited[i] = visited[x] + 1
                    q.append(i)

print(bfs())

---

실행 화면

 

---

채점 결과