Python/DFS BOJ-5014 스타트링크

Posted on | In

[Python/DFS] BOJ-5014 스타트링크

📌문제링크

S층에서 G층까지 버튼을 몇번 누르면 되는지 구하는 문제다.
S층과 G층이 동일할경우를 놓치지 말자.

BFS solution

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
import sys
from collections import deque
input = sys.stdin.readline

if __name__ == '__main__':
    F,S,G,U,D = map( int, input().split())
    queue = deque([S])
    visited = [False]*(1000001)
    visited[S]=1
    while queue :
        s = queue.popleft()
        if s == G :
            break
        ns = s + U # 올라가는 경우
        if U > 0 and 1<=ns<=F :
            if not visited[ns] :
                queue.append(ns)
                visited[ns] = visited[s]+1
        ns = s - D # 내려가는 경우
        if D > 0 and 1<=ns<=F:
            if not visited[ns] :
                queue.append(ns)
                visited[ns] = visited[s]+1
    if visited[G] == False:
        print('use the stairs')
    else : 
        print(visited[G]-1)