[백준], 1697번, 숨바꼭질 (파이썬)

문제

수빈이는 동생과 숨바꼭질을 하고 있다. 수빈이는 현재 점 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

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

def dfs():
    visited = [0]*100001 #방문했는지 확인
    dx = [-1,1,0]
    dy = [1,1,2]
    queue = deque([n])
    if n==k:
        return 0
    while not visited[k] and queue:
        move = queue.popleft()
        for i in range(3):
                next_move = move*dy[i] + dx[i]
                if 0<=next_move<100001 and not visited[next_move]:
                    visited[next_move] = visited[move] + 1
                    queue.append(next_move)
    return visited[k]

print(dfs())

#재귀 방법으로 풀기

def c(n,k):
    if n>=k:
        return n-k
    elif k == 1:
        return 1
    elif k%2:
        return 1+min(c(n,k-1),c(n,k+1))
    else:
        return min(k-n, 1+c(n,k//2))
    
n, k = map(int,input().split())
print(c(n,k))

문제해설

경우의 수가 정해져있어서 재귀적으로 풀면 더 빠르게 풀 수 있다.

하지만, BFS 유형을 풀다가 보면 생각하는 것은 쉽지 않다.