IT/알고리즘
[BOJ]1874.스택 수열
어센트
2020. 10. 7. 22:27
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
처음에 무한루프를 걸고 탈출하는 방식으로 문제를 해결하려고 했는데 그렇게 되면 NO를 출력시키는 부분의 예외처리를 해줄 수 없었다. 그래서 while 문에 조건문을 넣고 cnt가 num 이상이 될 때 까지 증가시킬 때 까지 반복했다.
만약 반복문을 탈출했을 때 stack 의 top이 num아니라면 NO를 출력시켜줘야한다.
풀이
n = int(input())
lst = []
cnt = 1
stack = []
flag = True
for i in range(n):
num = int(input())
while cnt <= num:
stack.append(cnt)
lst.append('+')
cnt += 1
if stack[-1] == num:
stack.pop()
lst.append('-')
else:
flag = False
if flag:
for i in lst:
print(i)
else:
print('NO')