코테 공부

백준[1874] 스택 수열 (파이썬 풀이)

moonsun623 2021. 10. 4. 00:19
반응형

https://www.acmicpc.net/problem/1874

 

1874번: 스택 수열

1부터 n까지에 수에 대해 차례로 [push, push, push, push, pop, pop, push, push, pop, push, push, pop, pop, pop, pop, pop] 연산을 수행하면 수열 [4, 3, 6, 8, 7, 5, 2, 1]을 얻을 수 있다.

www.acmicpc.net

 

내가 한 풀이

n=int(input())
count=1
stack=[]
answer=[]
for i in range(1,n+1):
    data=(int(input()))
    while count<=data :
        stack.append(count)
        count+=1
        answer.append('+')
    if stack[-1]==data:
        stack.pop()
        answer.append('-')
    else:
        print("NO")
        exit(0)
print('\n'.join(answer)) #answer에 있는 요소 한줄 씩 출력