알고리즘/백준
[백준 17298][파이썬] 오큰수
2twix2
2022. 7. 5. 15:15
https://www.acmicpc.net/problem/17298
17298번: 오큰수
첫째 줄에 수열 A의 크기 N (1 ≤ N ≤ 1,000,000)이 주어진다. 둘째 줄에 수열 A의 원소 A1, A2, ..., AN (1 ≤ Ai ≤ 1,000,000)이 주어진다.
www.acmicpc.net
import sys
input = sys.stdin.readline
N = int(input())
num_list = list(map(int, input().strip().split()))
stack = []
result = ['-1' for _ in range(N)]
for i in range(0, N):
while len(stack) != 0 and num_list[stack[-1]] < num_list[i]:
result[stack.pop()] = str(num_list[i])
stack.append(i)
print(' '.join(result))