[코딩 테스트 연습(파이썬/Python)] 백준(BOJ) 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의 위치로 이동하게 된다.

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

 

[입력]

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

[출력]

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

 

[아이디어]

1) 출발 지점에 갈 수 있는 곳들을 queue에 넣으며 이동한다.

2) 일종의 메모이제이션을 활용하면서 중복된 연산을 피한다.

 

from collections import deque
n,k = map(int,input().split())
cnt = {n:0}
q = deque([n])
while q:
    now = q.popleft()
    if now==k:
        break
    if (t:=2*now)<=100000:
        if t in cnt:
            if cnt[t] > cnt[now]:
                cnt[t] = cnt[now]
                q.append(t)
        else:
            cnt[t] = cnt[now]
            q.append(t)
    for t in [now-1, now+1]:
        if 0<=t<=100000:
            if t in cnt:
                if cnt[t] > cnt[now]+1:
                    cnt[t] = cnt[now]+1
                    q.append(t)
            else:
                cnt[t] = cnt[now]+1
                q.append(t)
print(cnt[k])

재귀함수로도 해결할 수 있는 문제다.

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

def f(n,k):
    if n>=k:
        return n-k
    elif n==0:
        return f(1,k) + 1
    elif k%2==0:
        return min(f(n,k//2),k-n)
    else:
        return min(f(n,(k+1)),f(n,(k-1)))+1
print(f(n,k))