백준 17298

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

17298번: 오큰수

<내 풀이>

N = int(input())
A = list(map(int, input().split()))

answer = [-1] * N  # 결과 배열을 -1로 초기화
stack = []

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

print(*answer)

> stack에는 따로 i를 저장함으로써 현재 값과 비교해야할 값들의 index 순번을 꺼내올 수 있습니다. 만약 현재 값이 이전 값보다 크다면, 이전 값들의 순번을 계속 꺼내보면서 현재값이 큰 동안 answer 배열의 값을 현재값으로 변경해줍니다.