백준 python 기록

백준 파이썬 17298 오큰수

우히힝 2022. 4. 12. 23:19

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

 

 

for문을 돌면서 브루트포스 풀기에는 크기가 컸고 스텍으로 푸는게 가장 현명하다고 생각하였습니다.

 

FOR 문을 돌면서 크기를 체크하고 조건에 해당하면 STACK에 있는 변수값으로 결과값을 도출하면 됩니다.

n = int(input())
s = list(map(int,input().split()))
answer = [-1] * n
stack = []

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


print(*answer)

 

상당히 쉬운데 고난이도에 위치하고있습니다.

발상의 전환이 필요한 문제인거같네요