알고리즘/백준

[백준] 1874번 스택 수열 (파이썬)

알감자 2021. 12. 28. 20:39
 

1874번: 스택 수열

1부터 n까지에 수에 대해 차례로 [push, push, push, push, pop, pop, push, push, pop, push, push, pop, pop, pop, pop, pop] 연산을 수행하면 수열 [4, 3, 6, 8, 7, 5, 2, 1]을 얻을 수 있다.

www.acmicpc.net

 

코드

n = int(input())

stack = []
count = 1
temp = True
answer = []
for i in range(n):
    num = int(input())
    while count <= num:
        stack.append(count)
        answer.append('+')
        count += 1
    if stack[-1] == num:
        stack.pop()
        answer.append('-')
    else:
        temp = False

if temp == False:
    print('NO')
else:
    for i in answer:
        print(i)