QA & Engineering Blog

A Blog about Quality · Automation · Engineering

🏠 홈으로

[Gold V] 숨바꼭질 3 - 13549

문제 링크

성능 요약

메모리: 117788 KB, 시간: 180 ms

분류

0-1 너비 우선 탐색, 너비 우선 탐색, 데이크스트라, 그래프 이론, 그래프 탐색, 최단 경로

제출 일자

2024년 1월 1일 23:17:17

문제 설명

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

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

입력

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

출력

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

💡 Solutions

📄 숨바꼭질 3.py


import sys
from collections import deque
N, K = map(int, sys.stdin.readline().split())
# https://www.acmicpc.net/board/view/38887#comment-69010

def bfs(start, end):
    ans = 0
    
    que = deque([(start, 0)])
    
    target = max(start, end)*2 + 1
    
    matrix = [0 for _ in range(target)] 
    is_travel = [False for _ in range(target)] 
    
    is_travel[start] = True
    
    while que:
        now, depth = que.popleft()
        
        if now == end : return matrix[now]
        # print(now, depth)
        for i in (2, -1, 1):
            
            if i == 2 and now <= target//2:
                _next = 2*now
                if is_travel[_next] : continue
                matrix[_next] = depth
                que.appendleft((_next, depth))
                is_travel[_next] = True
                
            if i == -1 and now > 0:
                _next = now-1
                if is_travel[_next] : continue
                matrix[_next] = depth + 1
                que.append((_next, depth + 1))
                is_travel[_next] = True
                
            if i == 1 and now <= target//2:
                _next = now+1
                if is_travel[_next] : continue
                matrix[_next] = depth + 1
                que.append((_next, depth + 1))
                is_travel[_next] = True
            
            # print(matrix)
            if _next == end : return matrix[_next]
        # print()
    
    return ans

print(bfs(N,K))