수빈이는 동생과 숨바꼭질을 하고 있다. 수빈이는 현재 점 N(0 ≤ N ≤ 100,000)에 있고, 동생은 점 K(0 ≤ K ≤ 100,000)에 있다. 수빈이는 걷거나 순간이동을 할 수 있다. 만약, 수빈이의 위치가 X일 때 걷는다면 1초 후에 X-1 또는 X+1로 이동하게 된다. 순간이동을 하는 경우에는 1초 후에 2*X의 위치로 이동하게 된다.
수빈이와 동생의 위치가 주어졌을 때, 수빈이가 동생을 찾을 수 있는 가장 빠른 시간이 몇 초 후인지 구하는 프로그램을 작성하시오.
첫 번째 줄에 수빈이가 있는 위치 N과 동생이 있는 위치 K가 주어진다. N과 K는 정수이다.
수빈이가 동생을 찾는 가장 빠른 시간을 출력한다.
5 17
4
from collections import deque
n,m=map(int,input().split())
vec=[1,-1,2]
queue=deque([(n+1,1),(n-1,1),(n*2,1)])
while queue:
loc,sec=queue.popleft()
if loc==m:
break
for v in vec:
if v!=2:
queue.append((loc+v,sec+1))
else:
queue.append((loc*2,sec+1))
print(sec)
메모리 초과
가 발생했다.from collections import deque
n,m=map(int,input().split())
vec=[1,-1,2]
queue=deque([(n,0)])
while queue:
loc,sec=queue.popleft()
if loc==m:
print(sec)
break
for v in vec:
if v!=2:
if 0<loc+v<=m:
queue.append((loc+v,sec+1))
else:
if 0<loc*v<=m:
queue.append((loc*2,sec+1))
from collections import deque
n,m=map(int,input().split())
queue=deque([n])
visited=[0]*100001
while queue:
loc=queue.popleft()
if loc==m:
print(visited[loc])
break
for v in [loc+1,loc-1,loc*2]:
if 0<=v<=100000 and not visited[v]:
visited[v]=visited[loc]+1
queue.append(v)