STUDY/Algorithm
[백준] 13549 숨바꼭질 3 python
sinawi95
2022. 2. 7. 06:38
728x90
https://www.acmicpc.net/problem/13549
13549번: 숨바꼭질 3
수빈이는 동생과 숨바꼭질을 하고 있다. 수빈이는 현재 점 N(0 ≤ N ≤ 100,000)에 있고, 동생은 점 K(0 ≤ K ≤ 100,000)에 있다. 수빈이는 걷거나 순간이동을 할 수 있다. 만약, 수빈이의 위치가 X일
www.acmicpc.net
BFS로 풀수있는 쉬운 문제이다.
from collections import deque
def main():
n, k = map(int, input().split())
memo = [200000 for _ in range(100001)]
q = deque()
q.append(n)
memo[n] = 0
answer = None
while q:
cur = q.popleft()
if cur == k:
answer = memo[cur]
break
if 2 * cur <= 100000:
if memo[2 * cur] > memo[cur]:
memo[2 * cur] = memo[cur]
q.append(2 * cur)
if cur - 1 >= 0:
if memo[cur - 1] > memo[cur]:
memo[cur - 1] = memo[cur] + 1
q.append(cur - 1)
if cur + 1 <= 100000:
if memo[cur + 1] > memo[cur]:
memo[cur + 1] = memo[cur] + 1
q.append(cur + 1)
print(answer)
if __name__ == "__main__":
main()