13549번: 숨바꼭질 3
수빈이는 동생과 숨바꼭질을 하고 있다. 수빈이는 현재 점 N(0 ≤ N ≤ 100,000)에 있고, 동생은 점 K(0 ≤ K ≤ 100,000)에 있다. 수빈이는 걷거나 순간이동을 할 수 있다. 만약, 수빈이의 위치가 X일
www.acmicpc.net
문제
수빈이는 동생과 숨바꼭질을 하고 있다. 수빈이는 현재 점 N(0 ≤ N ≤ 100,000)에 있고, 동생은 점 K(0 ≤ K ≤ 100,000)에 있다. 수빈이는 걷거나 순간이동을 할 수 있다. 만약, 수빈이의 위치가 X일 때 걷는다면 1초 후에 X-1 또는 X+1로 이동하게 된다. 순간이동을 하는 경우에는 0초 후에 2*X의 위치로 이동하게 된다.
수빈이와 동생의 위치가 주어졌을 때, 수빈이가 동생을 찾을 수 있는 가장 빠른 시간이 몇 초 후인지 구하는 프로그램을 작성하시오.
입력
첫 번째 줄에 수빈이가 있는 위치 N과 동생이 있는 위치 K가 주어진다. N과 K는 정수이다.
출력
수빈이가 동생을 찾는 가장 빠른 시간을 출력한다.
코드
- 0-1 BFS
from collections import deque
n, k = map(int, input().split())
q = deque([[n, 0]])
visited = [False] * 100_001
visited[n] = True
while q:
now, cnt = q.popleft()
if now == k:
print(cnt)
break
if 0 <= now * 2 <= 100_000 and not visited[now * 2]:
q.appendleft([now * 2, cnt])
visited[now * 2] = True
for i in [now - 1, now + 1]:
if 0 <= i <= 100_000 and not visited[i]:
q.append([i, cnt + 1])
visited[i] = True
- 다익스트라
import heapq
n, k = map(int, input().split())
visited = [False] * 100_001
q = [(0, n)]
while True:
cnt, now = heapq.heappop(q)
if now == k:
print(cnt)
break
visited[now] = True
if now * 2 <= 100_000 and not visited[now*2]:
heapq.heappush(q, (cnt, now*2))
for i in [now-1, now+1]:
if 0 <= i <= 100_000 and not visited[i]:
heapq.heappush(q, (cnt+1, i))
'문제풀이 > DFS_BFS' 카테고리의 다른 글
[Python/파이썬] 백준 14620번 꽃길 (0) | 2023.03.01 |
---|---|
[Python/파이썬] 백준 9079번 동전 게임 (2) | 2023.02.28 |
[Python/파이썬] 백준 2667번 단지번호붙이기 (0) | 2023.02.25 |
[Python/파이썬] 백준 2178번 미로 탐색 (0) | 2023.02.25 |
[Python/파이썬] 백준 1325번 효율적인 해킹 (0) | 2023.02.24 |