BAEKJOON #13549 숨바꼭질 3 (BFS, 다익스트라) - python

nathan·2021년 11월 1일
0

알고리즘문제

목록 보기
82/102

숨바꼭질 3

출처 : 백준 #13549

시간 제한메모리 제한
2초512MB

문제

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

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


입력

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


출력

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


입출력 예시

예제 입력 1

5 17

예제 출력 1

2


힌트

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


풀이

생각

  • 최단경로를 구하는 문제이다.
  • BFS, 다익스트라를 이용하여 풀 수 있다.
  • 이 문제의 경우 위의 문제 유형을 고려하지 않고 그냥 머릿속에서 생각나는 대로 코드를 짜보았다.
  • 우선 visited를 두어 방문 여부를 체크한다. (중복되는 경우를 방지하기 위함)
  • deque에 현재 위치(now)와 초(sec)를 튜플로 묶어 넣어준다.
  • 목적지(m)보다 현재 위치(now)가 큰 경우에는 빼는 경우만 존재하므로 따로 분기하여 주었다.

python code

# 백준 13549번 숨바꼭질 3
from sys import stdin
from collections import deque
input = stdin.readline

n, m = map(int, input().split())
visited = [0] * (10**5+1)
visited[n] = 1
temp = deque([(n, 0)])
sec = -1
while temp:
    now, sec = temp.popleft()
    if now == m:    # 목적지 도착시
        break
    else:
        if now > m: # 만약 목적지보다 현재 위치가 더 가있는 경우 (빼기만 해야함)
            if 0 <= now-1 and visited[now-1] == 0:   # 
                temp.append((now-1, sec+1))
                visited[now-1] = 1
        else:   # 목적지에 도달하지 못한 나머지 경우 
            # 곱하기 2의 경우가 먼저 와야하는 이유 : 0초의 시간이 걸리기 때문
            if now != 0 and now*2 <= 10**5 and visited[now*2] == 0:
                temp.append((now*2, sec))
                visited[now*2] = 1
            if 0 <= now-1 and visited[now-1] == 0:
                temp.append((now-1, sec+1))            
                visited[now-1] = 1
            if now+1 <= m and visited[now+1] == 0:
                temp.append((now+1, sec+1))
                visited[now+1] = 1

print(sec)
profile
나는 날마다 모든 면에서 점점 더 나아지고 있다.

0개의 댓글