[BAEKJOON] 1697번: 숨바꼭질 - 파이썬(python)

문제

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

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

입력

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

출력

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

풀이 💡

기본적으로 문제를 딱 봤을 때, 현재 위치 N에서 N-1, N+1, N*2 를 돌면서 BFS 동작을 하리란 건 알 수 있었다. 

 

위와 같이 BFS를 통해 각각 뻗어 나가고, 만약 N == K 가 된다면, 이동한 거리를 출력하면 된다.

(DFS를 한다면, 아마 끝도 없이 밑으로 내려갈 것으로 생각된다.)

 

어려웠던 부분은 역시 구현이다. 사실 생각해보면 엄청 어려운 문제는 아닌 것 같은데,, 한 번에 풀지 못해 아쉽다

 

먼저 거리를 계산하는건데, 처음에는 정말 단순하게 count를 생각했다. 

단순히 BFS 를 실행할 때마다 count를 해주는 것이다.. 

 

하지만 너무 단순한 생각이었고, 더 좋은 방법으로는 거리 리스트를 이용하는 것이다.

먼저 dist 리스트를 만든 후 0으로 초기화를 해준다. 

 

그다음, 한 사이클 (N-1, N+1, N*2)의 위치에서 같은 값을 갖는 것이다. 

예를 들어, 현재 위치가 5라면, 4,6,10의 위치에 모두 1을 표시하는 것이다. 

이렇게 하면 해당 위치로 갈 때마다 몇 번 이동을 해야 하는지 알 수 있게 된다. 

코드 💡

from collections import deque

def BFS():
    queue = deque()
    queue.append(N)
    while queue:
        present = queue.popleft()
        if present == K:
            print(dist[present])
            break
        for not_present in (present+1, present-1, present*2):
            if 0 <= not_present <= MAX and not dist[not_present]:
                dist[not_present] = dist[present] + 1
                queue.append(not_present)

MAX = 10 ** 5
dist = [0] * (MAX+1)
N, K = map(int, input().split())
BFS()