글 읽기 - 파이썬 Index Error

eric010506 2년 전 0

반례들은 거의 다 해본거같은데 이상하게 25%에서 index error가 발생하네요.. 이유를 전혀 모르겠습니다ㅜㅜ

import sys
num = int(sys.stdin.readline().rstrip())
num_stack = []
for i in range(num):
    num_stack.append(i+1)

num_stack.sort(reverse=True)

wanted_list = []

for i in range(num):
    wanted_list.append(int(sys.stdin.readline().rstrip()))

temp_stack = []
result_stack = []
oper_list = []

for i in range(wanted_list[0]):
        temp_stack.append(num_stack.pop())
        oper_list.append("+")

for i in wanted_list:
    if len(num_stack) and i >= num_stack[-1]:
        for j in range(i - num_stack[-1] + 1):
            temp_stack.append(num_stack.pop())
            oper_list.append("+")
    if len(temp_stack) and i <= temp_stack[-1]:
        for j in range(temp_stack[-1] - i + 1):
            result_stack.append(temp_stack.pop())
            oper_list.append("-")

if result_stack == wanted_list:
    for i in range(len(oper_list)):
        print(oper_list[i])
else:
    print("NO")

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

문제

문제

출처

ICPC

파이썬 Index Error

eric010506 2년 전 0

Baekjoon Online Judge

채점 현황

문제

유저 대회

고등학교 대회

출처

대학교 대회

도움말