[Algo] 백준 1697번 ‘숨바꼭질’ (python)

[Silver I] 숨바꼭질 - 1697

문제 링크

성능 요약

---

메모리: 35108 KB, 시간: 104 ms

분류

---

너비 우선 탐색, 그래프 이론, 그래프 탐색

문제 설명

---

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

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

입력

---

첫 번째 줄에 수빈이가 있는 위치 N과 동생이 있는 위치 K가 주어진다. N과 K는 정수이다.

출력

---

수빈이가 동생을 찾는 가장 빠른 시간을 출력한다.

소스 코드

---

from collections import deque

n, k = map(int, input().split())
dist = [0] * 100001
max = 100000

def bfs(n):
    queue = deque()
    queue.append(n)
    while len(queue) > 0:
        curr = queue.popleft()
        if curr == k:
            print(dist[curr])
            break
        for next in (curr+1, curr-1, curr*2):
            if 0 <= next <= max and dist[next] == 0:
                dist[next] = dist[curr] + 1
                queue.append(next)

bfs(n)

코드 리뷰

---

이 문제는 BFS로 풀어야 하는 문제다. 이 코드의 핵심이 되는 부분은

for next in (curr+1, curr-1, curr*2):
    if 0 <= next <= max and dist[next] == 0:
        dist[next] = dist[curr] + 1
        queue.append(next)

인데, 이는 현재 노드에서 갈 수 있는 노드 세개를 확인해보겠다는 뜻이다. 예를 들어서 현재 노드가 5라면, 다음에 갈 수 있는 노드는 4, 6, 10이므로 이 노드 세 개를 전부 확인하며 if 0 <= next <= max and dist[next] == 0조건에 맞는, 즉 다음 노드가 범위를 넘어서지 않으면서 아직 방문한 적 없는 노드일 때 dist[next] = dist[curr] + 1코드를 실행해 거리를 저장하도록 하고 queue에 넣어둔다.