[BOJ/백준] 1874번 스택 수열 | 파이썬

문제

풀이

import sys

n = int(input())
target = [int(sys.stdin.readline().strip()) for _ in range(n)]
stack = []
res = []
for i in range(1, n+1):
    stack.append(i)
    res.append('+')
    while stack and target:
        if stack[-1] == target[0]:
            stack.pop()
            res.append('-')
            del target[0]
        else:
            break
if target:
    print('NO')
else:
    for i in range(len(res)):
        print(res[i])

 

1부터 n까지의 수로 입력된 수열을 만드는 문제.

 

sys.stdin.readline().strip()을 사용해서 입력값을 target 변수에 저장했다.

 

이후 1부터 n까지의 수로 반복문을 만들었다.

• while문 내에서 stack과 target이 비어있지 않은 경우 stack[-1]과 target[0]값을 비교했다.
• 두 값이 일치하면 stack에서 값을 꺼내고, target[0]값을 제거했다.

 

res변수에는 push 연산일 때 '+'가, pop 연산일 때 '-'가 추가된다. 

* 최종 출력을 위해 res변수를 만들었다.

 

1부터 n까지의 수로 target을 만들 수 있는 경우 최종적으로 target값은 비게 된다(del target[0] 때문).

이를 이용해 target이 존재하는 경우 'NO'를, 존재하지 않는 경우 res값을 출력했다.

 

---

input()을 썼을 때랑 sys.stdin.readline()을 썼을 때 차이가 많이 난다😮

 

백준 1874번: 스택 수열

1874번: 스택 수열