백준 1697번 - 숨바꼭질 (파이썬/Python)

문제

수빈이는 동생과 숨바꼭질을 하고 있다. 수빈이는 현재 점 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
def bfs():
    Q = deque()
    Q.append(N)     # Q 초기값은 N을 넣어준다.


    while Q:        # Q가 존재하면 계속 반복
        v = Q.popleft() # Q의 0번째 값을 빼고 v로 선언

        if v == K:      # v가 구하고자 하는 K가 되면 해당 칸에 도달하기 위해 반복한 횟수를 출력하고 함수 종료
            print(visited[K])
            return

        for i in (v-1, v+1, 2*v):   # v값의 -1, +1, *2 해준 값을 순회하고, 그 값이 visited 범위 안인지, 방문한적이 없는지 확인하고
            if 0<= i < len(visited) and not visited[i]:
                Q.append(i)         # 방문한적 없다면 Q에 i를 넣어주고, visited에 그 전보다 +1 한 값을 넣어준다. (1초가 지났기 때문)
                visited[i] = visited[v]+1

N, K = map(int, input().split())
visited = [0] * 100001

bfs()

후기

처음에 이 문제는 DP로 해결해야 하는 줄 알았는데, 값이 안나와서 알고리즘 분류를 확인해보니 그래프로 해결해야 하는 문제였다......

유형이 많아지니 슬슬 어떻게 푸는 문제인지 고민해봐야 하는 시간이 늘어나기 시작했다.

 

dfs에 이어 bfs에 관련된 문제도 풀어봤는데, 큐에 대해 생소해서 어째서 bfs가 너비우선 탐색인지에 대해 이해하지 못하고 있었다.하지만 해당 문제와 싸피 수업을 들으며, 어째서 bfs에 큐를 이용하는지, dfs에 스택을 이용하는지 알게 됐다.또한 bfs는 여러개의 초기값을 넣어, 그 시작점을 기준으로 퍼져나가는 탐색 역시 가능하다는 것을 알게 됐다.