1874번: 스택수열 (구현, 스택) - Fastcampus

www.acmicpc.net/problem/1874

1874번: 스택 수열

import sys


def run(N, a):
    res = []
    i, st = 1, []  # 스택

    for x in a:
        while x >= i:
            st.append(i)
            res.append("+")
            i += 1
        if st[-1] == x:
            st.pop()
            res.append("-")
        else:
            return ["NO"]

    return res


N = int(input())
a = [int(sys.stdin.readline()) for _ in range(N)]

ans = run(N, a)
[print(x) for x in ans]  # print("\n".join(ans))