5014번 스타트링크 파이썬

문제

입력 , 출력

정답

import sys
from collections import deque

f, s, g, u, d = map(int, input().split())
visited = [0 for i in range(f+1)]
dq = deque()
dq.append((s, 0))
cnt = -1
while dq:
    a, b = dq.popleft()
    if a == g:
        cnt = b
        break
    if a+u <= f:
        if visited[a+u] == 0:
            dq.append((a+u, b+1))
            visited[a+u] = 1
    if a-d >= 1:
        if visited[a-d] == 0:
            dq.append((a-d, b+1))
            visited[a-d] = 1

if cnt == -1:
    print('use the stairs')
else:
    print(cnt)

설명

s - > 시작 층
g - > 목표 층
u - > 올라가는 버튼 층수
d - > 내려가는 버튼 층수
현재 층수 + u or - d 가 갈수있는 층 수 일경우에
dq에 삽입해서 현재층수가 목표층 즉 g 까지 달성할때까지 bfs를 돌림
만약 g층에 도달하지 못했을경우 use the stairs 를 출력
아닐경우에는 cnt , 사용한 버튼 횟수를 출력

후기

다 풀었는데 print를 이상한곳에 써놨어서 2번이나 눈치를 못챘었다..
bfs는 골드 4,5 를 풀수있게되어서 매우 뿌듯하다🚀🎇

좋은 웹페이지 즐겨찾기