[백준] [Python] 17298번 오큰수 - 스택

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

17298번: 오큰수

 

 

Python

import sys
n = int(sys.stdin.readline().rstrip())
arr = list(map(int, sys.stdin.readline().split()))

stack = []
ans = []
for i in range(n-1, -1, -1):
    if not stack:
        ans.append(-1)
    elif stack[-1] > arr[i]:
        ans.append(stack[-1])
    else:
        isEmpty = False
        while stack[-1] <= arr[i]:
            stack.pop()
            if not stack:
                isEmpty = True
                break
        ans.append(-1 if isEmpty else stack[-1])

    stack.append(arr[i])


for i in range(n-1, -1, -1):
    print(ans[i], end=" ")

 

추가 테스트 케이스

입력

8
1 2 3 4 3 3 4 1

 

출력

2 3 4 -1 4 4 -1 -1

 

 

풀이