백준 13913 파이썬 - 숨바꼭질4

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

13913번: 숨바꼭질 4

문제

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

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

코드

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

풀이

숨바꼭질문제에서 move리스트를 추가했다. 출력하기 전에 arr에 temp를 추가하고 move[temp] 즉, 그 전에 있던 경로를 계속 arr에 더해준다. 그리고 arr의 값을 순서를 거꾸로 해서 출력하면 된다.