[BOJ] 17298. 오큰수

BaekJoon 17298. 오큰수

* 문제의 저작권은 BOJ 및 문제를 만든 사람에게 있습니다.

 

 

 

 

 

[문제 접근]

문제 자체는 어렵지 않았는데 시간초과 해결하는게 어려웠다.

그래서 백준님꺼 참고해서 풀었다.

 

 

 

 

 

 

 

 

[코드]

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStreamWriter;
import java.util.ArrayList;
import java.util.Stack;
import java.util.StringTokenizer;

public class Main_17298 {
	public static void main(String[] args) throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

		int N = Integer.parseInt(br.readLine()); //A의 크기
		int[] a = new int[N];
		int[] ans = new int[N];
		
		
		StringTokenizer st = new StringTokenizer(br.readLine());
		for (int i = 0; i < N; i++) {
			a[i] = Integer.parseInt(st.nextToken());
		}
		
		Stack<Integer> stack = new Stack<>();
		stack.push(0);
		
		for (int i = 0; i < N; i++) {
			if (stack.isEmpty()) {
				stack.push(i);
			}
			
			while (!stack.isEmpty() && a[stack.peek()] < a[i]) {
				ans[stack.pop()] = a[i];
			}
			stack.push(i);
		}
		
		while (!stack.isEmpty()) {
			ans[stack.pop()] = -1;
		}
		
		for (int i = 0; i < ans.length; i++) {
			bw.write(ans[i] + " ");
		}
		bw.flush();
		
		
	}
}