[백준 12851] 숨바꼭질 2 (with Python / BFS)

문제링크

https://www.acmicpc.net/problem/12851

12851번: 숨바꼭질 2

문제설명

 

문제

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

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

입력

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

출력

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

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

예제 입력 1 

5 17

예제 출력 1 

4
2

문제풀이

처음 시도한 코드 (틀린 코드)

from collections import deque

N, K = map(int, input().split())
M = 100000
dist = [(0, 0) for _ in range(M+1)]

def BFS(start):
    queue = deque([start])

    while queue:
        cur = queue.popleft()

        if cur == K:
            print(dist[cur][0])
            print(dist[cur][1])
            return

        for next in (cur-1, cur+1, cur*2):
            if 0 <= next <= M:
                # 이전에 방문한 노드가 아니라면
                if not dist[next][0]:
                    queue.append(next)
                    dist[next] = (dist[cur][0] + 1, 1)
                # 이미 방문한적이 있다면, 더 최소시간으로 갱신
                else:
                    # 방문할 노드의 최소시간과 이미 저장된 최소시간이 같다면 방법수 1증가
                    if dist[next][0] == dist[cur][0] + 1:
                        dist[next] = (dist[next][0], dist[next][1] + dist[cur][1])
                    minValue = min(dist[next][0], dist[cur][0] + 1)
                    dist[next] = (minValue, dist[next][1])

BFS(N)

- 튜플로 처리했었는데 다시보니 코드자체도 굉장히 보기 불편하다...

- 그리고 dist[next] = (dist[cur][0] +1 , 1) 부분이 문제였다

=> 다시 생각하면 당연한 일인데... 1이 아니라 이전값인 dist[cur][1]를 그대로 가져가야한다.

- 최소시간으로 도달하는 방법수를 전부 세야하므로 K에 도달했다고 해서 바로 출력해버리면 안되고 BFS시행이 끝나고 나서 값에 접근해야함

 

핵심

- dist : 열이 2개인 이차원배열로 만든다.

  첫번째 인덱스 원소 : 최소시간을 의미

  두번째 인덱스 원소:  최소시간으로 거리에 도달하는 방법수를 의미

 cf. 처음엔 튜플로 묶었으나 튜플은 불변한 특성이 있기 때문에 값을 변경하기 위해서는 재할당으로 처리해야하는 번거로움이 있다 => 리스트로 묶은 이유.

- 최소시간으로 도달하는 방법수를 다 세야하므로 BFS()시행이 끝난후에 값을 조회하기.

- 숨바꼭질2는 걷기, 순간이동 모두 비용이 1초이므로, 이미 방문한 노드에 대해 최단시간 값 갱신을 할 필요가 없다.

   => BFS로 항상 최단시간 찾기 가능

  cf. 숨바꼭질3은 순간이동 비용이 0초이므로, BFS내에서도 최단시간 값 갱신이 필요하다.

 

코드

from collections import deque

N, K = map(int, input().split())
M = 100000
dist = [[-1, 0] for _ in range(M+1)] #첫번째 인덱스는 최소시간을 의미, 두번째는 최소시간으로 도달하는 방법수를 의미

def BFS(start):
    queue = deque([start])
    dist[start] = [0, 1]

    while queue:
        cur = queue.popleft()

        for next in (cur-1, cur+1, cur*2):
            if 0 <= next <= M:
                # 이전에 방문한 노드가 아니라면
                if dist[next][0] == -1:
                    queue.append(next)
                    dist[next] = [dist[cur][0] + 1, dist[cur][1]]
                # 이미 방문한적이 있다면, 더 최소시간으로 갱신
                else:
                    # 방문할 노드의 최소시간과 이미 저장된 최소시간이 같다면 방법수 1증가
                    if dist[next][0] == dist[cur][0] + 1:
                        dist[next][1] += dist[cur][1]
                    dist[next][0] = min(dist[next][0], dist[cur][0] + 1)
                    

BFS(N)
print(dist[K][0])
print(dist[K][1])