BOJ_1874 스택 수열

import sys
lambda input(): sys.stdin.readline().rstrip()

N = int(input())
# 리스트는 모두 pop을 할 수 있도록 역순으로 생성한다.
# 수열
seq = list(reversed([int(input()) for _ in range(N)]))
# push 할 값들 (1 ~ N)
nums = [i for i in range(N, 0, -1)]

# 스택
s = []
# 마지막에 출력할 마커 + / -
m = []

# 수열의 원소가 존재할 동안
while seq:
    # 스택에 원소가 존재하고
    # 수열의 마지막 원소와 스택의 마지막 원소가 같으면
    if s and seq[-1] == s[-1]:
        # 수열과 스택의 원소를 pop 한 뒤
        seq.pop()
        s.pop()
        # 마커에 -를 추가한다.
        m.append("-")
    # 같지 않고
    # push할 값이 남아있으면
    elif nums: 
        # nums에서 pop 해온 현재 값을 스택에 push 한 뒤
        s.append(nums.pop())
        # 마커에 +를 추가한다.
        m.append("+")
    # 더 push 할 값이 없으면
    else:
        # NO를 출력하고
        print("NO")
        # 반복을 빠져나온다.
        break
# 반복을 정상적으로 종료했을 경우
else:
    # 입력한 마커들을 줄로 구분해 출력한다.
    print("\n".join(m))

import sys
input = sys.stdin.readline

int(input())
p = map(lambda x: int(x.rstrip()), input())

def solution():
    stack, result, i = [], [], 1
    # 입력받은 수열의 각 원소에 대해
    for n in p:
        # 현재 값 i가 수열의 원소보다 커질 때까지
        while i <= n:
            # 스택에 현재 값 i를 push하고
            stack.append(i)
            # 결과에도 +를 push 한다.
            result.append("+")
            i += 1

        # 이후에 스택의 마지막 원소를 pop 했을 때
        # 현재 원솟값과 같지 않으면
        if stack.pop() != n:
            # NO를 반환한다.
            return "NO"
        # 같으면
        else:
            # 결과에 -를 push 한다.
            result.append("-")

    # 반복문이 끝나면 결과의 마커들을 줄로 구분해서 반환한다.
    return "\n".join(result)

# solution에서 반환받은 값을 출력한다.
print(solution())

p = map(lambda x: int(x.rstrip()), input())