알고리즘 문제 풀이/백준

[백준] 1697 BFS 숨바꼭질 파이썬 풀이

파란색 가운 2023. 11. 4. 22:43

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

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

 

1697번: 숨바꼭질

수빈이는 동생과 숨바꼭질을 하고 있다. 수빈이는 현재 점 N(0 ≤ N ≤ 100,000)에 있고, 동생은 점 K(0 ≤ K ≤ 100,000)에 있다. 수빈이는 걷거나 순간이동을 할 수 있다. 만약, 수빈이의 위치가 X일

www.acmicpc.net

입력

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

출력

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

예제 입력 1 복사

5 17

예제 출력 1 복사

4

힌트

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

import sys
# sys.stdin.readline()
from collections import deque
sys.setrecursionlimit(100000000)
import heapq
INF = int(1e9)

def BFS(graph,start,dest):
    queue = deque()
    queue.append(start) # 시작점 원소부터 일단 큐 담는다.
    while queue:
        v = queue.popleft()
        if v == dest:
            print(graph[v])
            break

        for j in (v-1,v+1,2*v):
            if 0<=j<=100000 and not graph[j]: # 범위 사이에 존재하고, 겹치지 않는 원소라면
                graph[j] = graph[v] + 1
                queue.append(j)
          
N,K = map(int,sys.stdin.readline().split())
graph = [0] * 100001
BFS(graph,N,K)
 

딱 어떻게 풀지 갈피가 잡혔는데

솔직히 못 풀었다

난 for i in graph[v]:

i  = graph[v] + 1 이렇게 해주고 싶었는데

이게 인덱스로서의 접근은 안된다고 한다(오류 뜸)

그래서 코드가 복잡해지다가 꼬였다

결론은 그냥 for j in (v-1,v+1,2*v)를 해줘도 된다고 하네요

파이썬이 간편하게 좋지만 그만큼 알고 있어야 하는 기능이 많아서 만만치 않은 것 같다.