[백준]5014번: 스타트링크

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

5014번: 스타트링크

---

 

bfs로 풀었다.

최초 좌표를 큐에 넣고, 큐에서 하나씩 꺼내서 위 또는 아래 층의 범위와 방문여부를 체크하였다.

 

최초의 좌표 s의 방문 여부 체크를 하지 않아서 자꾸 오답이 나왔었다.

 

---

 

python

from collections import deque

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

elevator = [0 for _ in range(f+1)]
q = deque()
q.append(s)
elevator[s] = 1 # 최초의 좌표 방문체크

if s == g:
  print(0)
  exit(0)

while q:
  cur = q.popleft()
  if cur == g:
    break
  for i in (u, -d):
    next = cur + i
    if 0 < next <= f and elevator[next] == 0:
      elevator[next] = elevator[cur] + 1
      q.append(next)

if elevator[g] == 0:
  print("use the stairs")
else:
  print(elevator[g]-1)