[백준] 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을 사용하여 접근해봤다. 

기본적인 BFS 함수

# BFS 함수 정의
def bfs(graph, start, visited):
    # 큐(Queue) 구현을 위해 deque 라이브러리 사용
    queue = deque([start])
    # 현재 노드를 방문 처리
    visited[start] = True
    # 큐가 빌 때까지 반복
    while queue:
        # 큐에서 하나의 원소를 뽑아 출력
        v = queue.popleft()
        print(v, end=' ')
        # 해당 원소와 연결된, 아직 방문하지 않은 원소들을 큐에 삽입
        for i in graph[v]:
            if not visited[i]:
                queue.append(i)
                visited[i] = True

위 코드에서 visited인 노드 방문 처리 리스트 대신, time이라는 리스트를 사용해서

해당 노드의 time이 0이면 방문하지 않은 노드로 생각하고 time에 걸린 시간(깊이)를 저장해줬다. 

답안 코드

from collections import deque
N, K = map(int, input().split(" "))

queue = deque([N])
time = [0]*100001

while queue:
    now = queue.popleft()
    if now == K:
        print(time[now])
        break
    for move in [now-1, now+1, 2*now]:
        if 0 <= move <= 100000:
            if not time[move]:
                time[move] = time[now] + 1
                queue.append(move)

결과