[BaekJoon] 12851 숨바꼭질2 (Gold 4) - Python

문제

 

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

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

---

입출력 예시

 

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

출력: 첫째 줄에 수빈이가 동생을 찾는 가장 빠른 시간을 출력한다.

둘째 줄에는 가장 빠른 시간으로 수빈이가 동생을 찾는 방법의 수를 출력한다.

 

입력예시1)

5 17

출력예시1)

4
2

---

코드

 

from collections import deque, defaultdict

N, K = map(int, input().split())
res = defaultdict(int)
visited = [False for _ in range(100001)]
q = deque()
q.append([N, 0])

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

for key in res.keys():
    print(key)
    print(res[key])
    break

---

실행 화면

---

채점 결과