Koala - 12기/코딩테스트 준비 스터디

[백준/python3] 1874번: 스택 수열

יוֹסֵף 2023. 10. 30. 06:33

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


문제


코드

import sys
input = sys.stdin.readline
n = int(input())
check = list()
y = 1
count = list()
flag = 0
for _ in range(n):
    num = int(input())
    while y <= num:
        check.append(y)
        count.append('+')
        y += 1
    if check[-1] == num:
        count.append('-')
        check.pop()
    else:
        flag = 1
        break
if flag == 0:
    print('\n'.join(count))
else:
    print("NO")