[알고리즘/백준] 1874: 스택 수열(python)

스택의 특성을 이해해야 푸는게 가능하다.

from sys import stdin

n = int(stdin.readline().strip())
a = list(int(stdin.readline().strip()) for _ in range(n))
stack, ans, cnt, flag = [], [], 1, False
for i in a:
    while cnt <= i:
        stack.append(cnt)
        ans.append('+')
        cnt += 1
    if stack[-1] == i:
        stack.pop()
        ans.append('-')
    else:
        flag = True
if flag:
    print('NO')
else:
    print(*ans, sep='\n')