[백준] 5014번 스타트링크 - 파이썬

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

5014번: 스타트링크

 

 

풀이

BFS로 모든경우의 수를 visit에 저장해준 뒤 target값을 출력

# 5014번 스타트링크 
from collections import deque

floor, start, target, u, d = map(int, input().split())
visit = [0] * (floor+1)
visit[0] = 1
dx = [u, -d]

def bfs():
    queue = deque()
    queue.append(start)
    visit[start] = 1
    while queue:
        x = queue.popleft()
        for i in range(2):
            nx = x + dx[i]
            if 0 <= nx <= floor and visit[nx] == 0:
                visit[nx] = visit[x] + 1
                queue.append(nx)
    return visit[target]

ans = bfs()

if ans == 0:
    print("use the stairs")
else:
    print(ans-1)

출력결과