[Python/파이썬] 백준 5014번: 스타트링크

문제

풀이

전형적인 너비우선 탐색 문제, heap을 이용해 풀어보았다.

 

코드

from sys import stdin
import heapq
input = stdin.readline

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

visited = [False] * (f + 1)
q = [(0, s)]
visited[s] = True
ans = "use the stairs"
while q:
      cnt, now = heapq.heappop(q)
      if now == g:
            ans = cnt
            break
      if now - d > 0:
            if not visited[now - d]:
                  visited[now - d] = True
                  heapq.heappush(q, (cnt + 1, now - d))
      if now + u <= f:
            if not visited[now + u]:
                  visited[now + u] = True
                  heapq.heappush(q, (cnt + 1, now + u))

print(ans)

 

마무리

비슷한 유형의 문제를 많이 풀어봐서 쉬웠다.