[백준] 1874번. 스택 수열 -Python

백준 1874번 - 스택 수열 문제

 

나의 막힌 점 : 문제 이해를 처음에 못했음. 이후에 수열 성립이 안될 때 조건을 찾기 어려웠음 -> 스택의 기본 원리를 잘 생각해보자

n = int(input())
arr = [int(input()) for _ in range(n)]

stack = []
current = 1
result = []

for num in arr:
    if current <= num:
        for j in range(current,num+1):
            stack.append(j)
            result.append("+")
            current += 1

    if stack[-1] == num:
        stack.pop()
        result.append("-")

    else:
        print("NO")
        exit(0)

for i in result:
    print(i)