[백준🥈1] #1697 숨바꼭질 (python)

https://www.acmicpc.net/problem/1697

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

 

✔️ Code

import collections
import sys
input = sys.stdin.readline

A, B = map(int,input().split())
queue = collections.deque() #양방향큐
queue.append((A, 0))
visited = [False] * 100001
result = 0

while queue:
    num, count = queue.popleft()
    if num == B:
        result = count
        break

    if 0 <= 2 * num <= 100000 and visited[2 * num] == False:
        queue.append((2 * num, count + 1))
        visited[2 * num] = True
    if 0 <= num + 1 <= 100000 and visited[num + 1] == False:
        queue.append((num + 1, count + 1))
        visited[num+1] = True
    if 0 <= num - 1 <= 100000 and visited[num-1] == False:
        queue.append((num - 1, count + 1))
        visited[num-1] = True


print(result)

 

✏️ Comment

bfs를 이용해 풀었다. 처음 시작점 N에서 갈 수 있는 점 N*2, N+1, N-1 세 점에서 또 각점이 갈 수 있는 세 점씩을 계속 탐색해서 K와 같을 때를 찾으면 된다. 단, count를 +1 해주는 위치와 사용하는 방법이 중요한데, count를 방문 안했을시 방문 할 때마다 +1 해주면 다른 탐색도 거쳐와서 N == K 일때도 생길 수 있기 때문에 count 값이 크게 나온다. 때문에 e라는 점이 큐에 몇번째로 들어갔는지 판단후 +1 해주기 위해 count를 q에 같이 넣어줌으로써 해결했다.

다시 풀어봐야한다는 게 함정.