백준 알고리즘 17298번 오큰수(python)

스택 문제이다. 처음에 스택에 인덱스 말고 수열의 원소를 넣는 방식으로 생각해서 못 풀었다.

결국 해답을 봤는데 한 끗 차이였다ㅠ 내일 다시 풀어봐야겠다.

from collections import deque

N = int(input())
nums = list(map(int, input().split()))
stack = deque()
res = [-1]*N
for i in range(N):
    while stack and nums[stack[-1]] < nums[i]:
        res[stack.pop()] = nums[i]
    stack.append(i)
print(*res)