[baekjoon] 백준 1697번(파이썬): 숨바꼭질

문제

1697번: 숨바꼭질

- 수빈이는 동생과 숨바꼭질을 하고 있다.

- 수빈이는 현재 점 N(0 ≤ N ≤ 100,000)에 있고, 동생은 점 K(0 ≤ K ≤ 100,000)에 있다.

- 수빈이는 걷거나 순간이동을 할 수 있다.

- 만약, 수빈이의 위치가 X일 때 걷는다면 1초 후에 X-1 또는 X+1로 이동하게 된다.

- 순간이동을 하는 경우에는 1초 후에 2*X의 위치로 이동하게 된다.

- 수빈이와 동생의 위치가 주어졌을 때, 수빈이가 동생을 찾을 수 있는 가장 빠른 시간이 몇 초 후인지 구하는 문제이다.

- 수빈이가 있는 위치 N과 동생이 있는 위치 K가 주어진다. N과 K는 정수이다.

알고리즘

- bfs 탐색을 통해 문제를 수행한다.

- 수빈과 동생이 갈 수 있는 모든 위치를 리스트로 표현한다.

- 현재 수빈에 위치와 동생에 위치가 같으면 멈춘다.

- 현재 수빈에 위치부터 이동할 수 있는 3가지 방법을 통해 이동한다.

- 각 방법으로 이동할 수 있고 한 번도 이동하지 않은 위치라면 현재 이동 시간 + 1을 해주어 현재 위치까지 가기 위한 시간을 초기화해준다.

- 다시 현재 바뀐 위치를 리스트에 추가하여 그 위치에서 다시 3가지 이동 방법을 수행한다.

코드

import sys
from collections import deque


# bfs 탐색
def bfs():

    queue = deque([n])

    # 큐에 탐색해야 하는 노드 없을 때까지 실행
    while queue:

        # 팝하여 탐색
        a = queue.popleft()

        # 현재 위치가 동생에 위치와 같다면 멈춘다.
        if a == k:
            print(graph[a])
            break

        # 3가지 이동 방법을 수행한다.
        for i in (a - 1, a + 1, a * 2):
            # 이때 정해진 위치로 이동했고 이동하지 않은 위치라면
            # 현재 이동시간 + 1 을 하여 현재 위치까지 가기 위한 시간을 초기화한다.
            # 그리고 다시 리스트에 추가하여 그 위치에서 다시 이동한다.
            if 0 <= i <= 100000 and not graph[i]:
                graph[i] = graph[a] + 1
                queue.append(i)


n, k = map(int,sys.stdin.readline().split())
graph = [0] * 100001
bfs()

github

junjange/CodingTest