출처 | 숨바꼭질[백준 1697]
수빈이는 동생과 숨바꼭질을 하고 있다. 수빈이는 현재 점 N(0 ≤ N ≤ 100,000)에 있고, 동생은 점 K(0 ≤ K ≤ 100,000)에 있다. 수빈이는 걷거나 순간이동을 할 수 있다. 만약, 수빈이의 위치가 X일 때 걷는다면 1초 후에 X-1 또는 X+1로 이동하게 된다. 순간이동을 하는 경우에는 1초 후에 2*X의 위치로 이동하게 된다.
수빈이와 동생의 위치가 주어졌을 때, 수빈이가 동생을 찾을 수 있는 가장 빠른 시간이 몇 초 후인지 구하는 프로그램을 작성하시오.
첫 번째 줄에 수빈이가 있는 위치 N과 동생이 있는 위치 K가 주어진다. N과 K는 정수이다.
수빈이가 동생을 찾는 가장 빠른 시간을 출력한다.
from sys import stdin
from collections import deque
def bfs(N, K):
dq = deque()
dq.append(N)
visited[N] = True
count = 0
while dq:
count += 1
for _ in range(len(dq)):
data = dq.popleft()
if data - 1 >= 0 and visited[data - 1] == False:
visited[data - 1] = True
if data - 1 == K:
return count
else:
dq.append(data - 1)
if data + 1 <= 100000 and visited[data + 1] == False:
visited[data + 1] = True
if data + 1 == K:
return count
else:
dq.append(data + 1)
if data*2 <= 100000 and visited[data * 2] == False:
visited[data * 2] = True
if data * 2 == K:
return count
else:
dq.append(data * 2)
return count
N, K = map(int, stdin.readline().split())
visited = [False] * 100001
if N == K:
print(0)
else:
print(bfs(N, K))
Olympiad > USA Computing Olympiad > 2006-2007 Season > USACO US Open 2007 Contest > Silver 2번