백준 5014 스타트링크 (파이썬)

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

5014번: 스타트링크

현재 위치에서 주어진 조건에 맞춰 갈 수 있는 모든 경로를 이동하면서 G층에 도착하면 그동안 누른 버튼 수를 출력

import sys
from collections import deque

input = sys.stdin.readline

def bfs():
    q.append(s-1)
    c[s-1] = 1
    while q:
        x = q.popleft()
        if x == g-1:
            print(c[x]-1)
            return
        if x+u < f and not c[x+u]:
            q.append(x+u)
            c[x+u] = c[x] + 1
        if x-d >= 0 and not c[x-d]:
            q.append(x-d)
            c[x-d] = c[x] + 1

    print("use the stairs")

f, s, g, u, d = map(int, input().split())
c = [0 for _ in range(f)]
q = deque()

bfs()