[백준/python]1874번 스택 수열

1874번: 스택 수열 (acmicpc.net)

1874번: 스택 수열

문제 분석

스택 구조로 자료를 저장하여 다시 꺼낸 수로 수열을 만들때 필요한 연산을 계산하는 문제이다.

소스 코드

n=int(input())
arr=[]
for i in range(n):
    a=int(input())
    arr.append(a)
stack=[]
s=[]
c=[]
t=1
for i in range(2*n):
    if stack==[]:
       stack.append(t)
       c.append("+")
       t+=1
    else:
        if stack[-1]==arr[0]:
            arr.pop(0)
            s.append(stack.pop())
            c.append("-")
        else:
            stack.append(t)
            c.append('+')
            t+=1

if len(s)!=n:
    print('NO')
else:
    print("\n".join(c))