🟩 [백준] [Python] [Gold4] 17298번_오큰수

문제

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

17298번: 오큰수

풀이

# 백준 17298번 문제 - 오큰수
import sys
input = sys.stdin.readline
n = int(input())
a = list(map(int, input().split()))
answer = [-1] * n 
stack = []

for i in range(n):
    while stack and a[stack[-1]] < a[i]:
        answer[stack.pop()] = a[i]
    stack.append(i)

print(*answer)

stack을 활용한 자료구조 문제이다. 전에 풀었던 탑과 매우 유사한 문제이다.