[BaekJoon] 1697 숨바꼭질 (Silver 1) - 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

---

코드

from collections import deque # deque 사용을 위해 import

cnt = [0] * 100001 # 미리 100000개의 인덱스 생성해두기

def sol(n): # bfs함수
    q = deque([n]) # deque에 n대입
    while q: # q가 비어있지 않다면
        n = q.popleft() # n = q의 왼쪽에서 원소 pop하여 대입
        if n == K: return cnt[n] # 만약 n == K라면 cnt리스트의 n번째 인덱스 값 전달
        for i in (n - 1, n + 1, 2 * n): # 튜플의 값 n - 1, n + 1, n * 2를 하나씩 가져와
            if 0 <= i < 100001 and not cnt[i]: # 만약 cnt[i]가 0이고, i의 값이 0 ~ 100000사이라면
                cnt[i] = cnt[n] + 1 # cnt[i]의 값은 cnt[n] + 1이다. (방문 번째 횟수: 그러니까 i번째 인덱스를 몇 번째로 방문했는지)
                q.append(i) # q에 i대입

N, K = map(int, input().split()) # N, K입력

result = sol(N) # bfs결과를 result에 대입
print(result) # result 출력

---

실행 화면

---

채점 결과