[ProblemSolving] 백준 - 13549 숨바꼭질3(dfs&bfs)

문제 링크

문제 설명

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

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

입력

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

출력

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

예제입력1

5 17

예제출력1

2

나의 풀이

인접노드(다음노드)로 이동할 때 마다 분기, 가장 빠른 시간(최단거리) => bfs로 해결

check[] 를 이용해 방문처리 & 최소시간누적을 동시에 한다.

순간이동 하는 if문을 선두에 배치한다.

그 이유는 n=2, k=4일때 count=0이 되어야 하는데, 걷는 if문이 순간이동 if문보다 먼저 나오면 오답이 나온다.

코드

from collections import deque

n, k = map(int, input().split())
max_size = 1000001
 
def bfs():
    queue = deque()
    queue.append(n)
    check = [-1]* max_size # 초기화
    check[n]= 0 
    while queue:
        v = queue.popleft()
        
        if v == k:
            return check[v]
        
        if 0<= v*2 < max_size and check[v*2] == -1:
            check[v*2] = check[v] # 방문했음 표시
            queue.append(v*2)
            
        for i in (v-1, v+1):
            if 0<= i < max_size and check[i] == -1:
                check[i] = check[v]+1
                queue.append(i)
print(bfs())