알고리즘/BOJ

[백준 / 파이썬] 1697. 숨바꼭질

감싹이 2023. 3. 26. 13:35

🎉🎉 문제

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

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

🎉🎉 풀이

☑️ BFS로 X-1, X+1, X*2한 위치를 넣으면서 최소 시간 탐색하는 문제

☑️ 인덱스 범위 고려하지 않으면 런타임에러 날 수 있음

 

🎉🎉 코드

from collections import deque

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

queue = deque()

#시작 위치와 시간을 큐에 추가
queue.append([n, 0])
visited = [False]*(100001)

answer = 0
while queue:
  n, time = queue.popleft()
  visited[n] = True

  #도착 위치에 도달했을 경우 시간 갱신 해주고 break
  if n == k:
    answer = time
    break

  
  #1칸 뒤로 이동
  if n-1 >= 0 and not visited[n-1]:
    queue.append([n-1, time+1])
  #1칸 앞으로 이동
  if n+1 <= 100000 and not visited[n+1]:
    queue.append([n+1, time+1])
  #순간이동으로 2배 앞으로 이동
  if n*2 <= 100000 and not visited[n*2]:
    queue.append([n*2, time+1])

#최소시간 출력
print(answer)