백준 13549 파이썬 - 숨바꼭질 3

https://www.acmicpc.net/problem/13549

13549번: 숨바꼭질 3

문제

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

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

코드

import sys
from collections import deque
n, k = map(int, sys.stdin.readline().split())
max_n = 100001
distance = [-1 for _ in range(max_n)]
distance[n] = 0
def bfs():
    q = deque()
    q.append(n)
    while q:
        a = q.popleft()
        if a == k:
            print(distance[a])
            exit()
        if 0 < a * 2 < max_n and distance[a * 2] == -1:
            distance[a * 2] = distance[a]
            q.appendleft(a * 2)
        if 0 <= a + 1 < max_n and distance[a + 1] == -1:
            distance[a + 1] = distance[a] + 1
            q.append(a + 1)
        if 0 <= a - 1 < max_n and distance[a - 1] == -1:
            distance[a - 1] = distance[a] + 1
            q.append(a - 1)
bfs()

풀이

예전 풀이처럼 for문으로 a * 2, a - 1, a + 1을 돌면 계속 정답처리가 안되서 다른 방식으로 풀었다. 주의할 점은 a * 2의 경우 전에 있던 시간을 적용시키면 되고 나머지 경우에는 1씩 증가시키면 된다. 부등호도 조심하자.