백준 5014. 스타트링크

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

5014번: 스타트링크

from collections import deque
F, S, G, U, D = map(int,input().split())
visited = [0]*(F+1)
def bfs(S):
    q = deque([S])
    visited[S] = 1  # 시작점 방문처리 해줘야 U나 D가 0이어도 눌렀다고 카운트하지 않음
    while q:
        t = q.popleft()
        if t == G:
            return visited[t]-1 # 시작점도 방문처리 해줬으므로 1 빼주기
        for w in [t+U,t-D]:
            if 1<=w<=F and not visited[w]: # 건물은 1층~F층임
                q.append(w)
                visited[w] = visited[t]+1
    return 'use the stairs'

print(bfs(S))