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

https://www.acmicpc.net/problem/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는 정수이다.

출력

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

예제 입력 1 복사

5 17

예제 출력 1 복사

4

힌트

수빈이가 5-10-9-18-17 순으로 가면 4초만에 동생을 찾을 수 있다.

 

 

일차원 배열을 이용하는 간단한 BFS문제입니다.

+1,-1,*2 로 파생 노드 종류를 설정하고, k와 일치하는 수를 찾았을 때의 노드레벨(탐색시간)을 출력하면 됩니다.

 

아마 많은 분들이 방문처리와 노드 설정 똑같이 했는데 4% 정도 채점됐을 때 틀렸다고 나오는 경우가 있을 것 같습니다.

저는 Flag변수를 설정해서 해결했는데요.

 

프로그래머스나 solution 함수를 작성해서 문제를 푸는 플랫폼이었으면 그냥 return 해버리면 되지만,

백준이나 SWEA의 경우 답이 나왔을 때 나머지 케이스들에 대해서도 살펴보는 경우가 있기 때문에

함수에서 return을 사용하는 것처럼 작동하기 위해 flag 변수를 설정했습니다.

 

이렇게 되면 제가 원하는 결과가 나왔을 때 while문에서 해당 반복까지만 진행하고 이후의 결과에 대해선 살펴보지 않고 바로 종료를 하게 됩니다.

저는 dn이 k와 일치했을 때, 해당 반복 내에서 이 경우 말고 또 다른 경우가 나올 수 있을 거라고 생각했습니다.

 

 

코드

#백준 1697 숨바꼭질
from collections import deque

n,k = map(int,input().split())
if n==k:
    print(0)
else:
    q=deque()
    visited = [False]*100001
    visited[n]=True
    q.append((n,0))
    answer = 0
    flag = True
    
    while q and flag:
        n, cnt = q.popleft()
        d = [n+1,n-1,n*2]
        for i in range(3):
            dn = d[i]
            if 0<=dn<=100000:
                if visited[dn]==False:
                    if dn == k:
                        answer = cnt+1
                        flag = False
                        break
                    q.append((dn,cnt+1))
                    visited[dn] = True
    print(answer)