백준 12851 숨바꼭질 2 파이썬 [BFS 가짓수 구하기]

dasd412·2022년 6월 11일
0

코딩 테스트

목록 보기
47/71

문제 설명

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

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

입력 조건

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

출력 조건

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

둘째 줄에는 가장 빠른 시간으로 수빈이가 동생을 찾는 방법의 수를 출력한다.


전체 코드

import sys
from collections import deque

n, k = list(map(int, sys.stdin.readline().split()))

queue = deque()
# (위치 , 이동 시간)의 튜플
queue.append((n, 0))

fastest_time = sys.maxsize
count = 0

visited = [False] * 100001

while queue:
    position, time = queue.popleft()

    # 큐에서 빼낼 때 방문 체크를 하면, 최단 거리로 이동할 수 있는 여러 경우를 담아낼 수 있다.
    # 예를 들어, 1에서 3으로 이동하는 최단 경우의 수는 2개이다.   1 + 1 + 1  또는 1 * 2 + 1이다.
    visited[position] = True

    if position == k and fastest_time >= time:
        fastest_time = time
        count += 1

    if 0 <= position - 1 <= 100000 and not visited[position - 1]:
        queue.append((position - 1, time + 1))

    if 0 <= position + 1 <= 100000 and not visited[position + 1]:
        queue.append((position + 1, time + 1))

    if 0 <= position * 2 <= 100000 and not visited[position * 2]:
        queue.append((position * 2, time + 1))

print(fastest_time)
print(count)

느낀 점

단순 BFS가 아닌, BFS로 얻어낸 최단 거리의 가짓수를 구하는 문제다.
응용하기 좋은 문제인 것 같다.

참고로 큐 2번 돌려서 풀면 메모리 초과나더라.


참고

https://sorryhyeon.tistory.com/m/62

profile
아키텍쳐 설계와 테스트 코드에 관심이 많음.

0개의 댓글