[백준 5014번] 스타트링크

문제 보기

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

5014번: 스타트링크

 

문제 해결

방문한 노드는 방문 표시를 해주는 것에 유의할 것

 

 

 

소스 코드

import sys
from collections import deque
input = sys.stdin.readline
f, s, g, u, d = map(int, input().split())

dp = [0 for _ in range(f+1)]
visited = [0 for _ in range(f+1)]

def bfs(v):
    q = deque([v])
    visited[v] = 1
    while q:
        v = q.popleft()
        if v == g:
            return dp[g]
        for i in (v+u, v-d):
            if 0 < i <= f and not visited[i]:
                visited[i] = 1
                dp[i] = dp[v] + 1
                q.append(i)
    if dp[g] == 0:
        return "use the stairs"            

print(bfs(s))