[백준/Python] 5014 스타트링크

from _collections import deque

f, s, g, u, d= map(int, input().split())

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

def bfs():
    que = deque()
    que.append(s)
    visited[s] = 1

    while que:
        floor = que.popleft()
        if floor==g:
            return visited[g]-1
        #up
        if floor+u <= f and not visited[floor+u]: # 조건 주의
            que.append(floor+u)
            visited[floor+u] = visited[floor]+1
        #down
        if floor-d >= 1 and not visited[floor-d]:
            que.append(floor-d)
            visited[floor-d] = visited[floor]+1

    return "use the stairs"

print(bfs())

BFS를 이용하면 된다.

queue는 멀티스레드용 모듈이라 매우 느리고, 싱글스레드용으로는 collections.deque을 쓴다고 한다.