[백준] 13549번 - 숨바꼭질 3

📩 출처

문제

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

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

입력

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

출력

수빈이가 동생을 찾는 가장 빠른 시간을 출력한다.

👉 생각

• 단순히 bfs 돌리면 되는 문제이긴 하지만, 메모리와 시간초과를 신경써야 하는 문제였다. 먼저 visited를 꼭 설정해서 한번 방문한 곳을 또 방문하지 않도록 해야한다.
• 이동할 수 있는 3가지 방법에서 순서도 중요하다! *2로 이동하는 것이 비용이 0이기 때문에 가장 먼저 for문을 돌거나 append를 해줄 때 appendleft를 해줘서 비용이 0인 *2 연산을 최우선적으로 처리하도록 하는 것이 이번 문제에 포인트다!
• 또한 -1 을 +1 보다 먼저 해야 하는데 4,6 예시를 보면 4>>3>>6 비용 1이 최소이지만, 더하기부터 한다면 비용 2가 나오기 때문에 -1을 먼저 해줘야 한다.

from collections import deque
def bfs(num):
    q = deque([])
    q.append((num, 0))
    visited = [0] * 100001
    visited[num] = 1
    while q:
        number, time = q.popleft()
        if number == k:
            return time

        for i in (2 * number, number - 1, number + 1):
            if 0 <= i < 100001 and not visited[i]:
                visited[i] = 1
                if i == 2 * number:
                    q.append((i, time))
                else:
                    q.append((i,time+1))

n, k = map(int, input().split())

print(bfs(n))