글 읽기 - 파이썬 질문

tinea17 2년 전 0

visit 리스트 값이 -1이 아니면 방문했다 생각하고 코드 작성한건데 테케는 맞고 제출하면 틀렸다고 나옵니다ㅠ

방문여부를 체크해주는 리스트를 따로 하나 더 만들어줘서 해결하는 방법말고

리스트 하나만으로 해결할 수 있는 방법은 없는지 궁금합니다!

from collections import deque

F, S, G, U, D = map(int, input().split())

visit = [-1 for _ in range(F+1)]

def bfs(x):
    queue = deque()
    queue.append(x)
    visit[x] = 0

    while queue:
        x = queue.popleft()
        if x == G:
            break

        for next_step in (x+U, x-D):
            if 1 <= next_step <= F and not visit[next_step]:
                visit[next_step] = visit[x] + 1
                queue.append(next_step)

bfs(S)

if visit[G] != -1:
    print(visit[G])
else:
    print("use the stairs")

mugglim 2년 전 1

반례 드립니다.

input
5 2 1 1 1

answer
1

wrong answer
use the stairs

아래 소스의 not visit[next_step] 부분을 수정해보시길 바랍니다!!.

if 1 <= next_step <= F and not visit[next_step]:

tinea17 2년 전 0

해결했습니다 감사합니다!

댓글을 작성하려면 로그인해야 합니다.

문제

문제

출처

ICPC

파이썬 질문

tinea17 2년 전 0

mugglim 2년 전 1

tinea17 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말