[JAVA] 백준 17298번 - 오큰수

📖 문제

 

📋 코드

import java.util.*;

public class Main {
	public static void main(String[] args) {
		Scanner sc = new Scanner(System.in);
		StringBuffer sb = new StringBuffer();
		int N = sc.nextInt();
		int A[] = new int[N + 1];
		int ans[] = new int[N + 1];
		Stack<Integer> st = new Stack<>();

		for (int i = 0; i < N; i++) {
			A[i] = sc.nextInt();
		}

		for (int i = N - 1; i >= 0; i--) {
			while (!st.empty() && st.peek() <= A[i]) {
				st.pop();
			}
			
			if(st.empty()) ans[i]=-1;
			else ans[i] = st.peek();
			
			st.push(A[i]);
		}

		for (int i = 0; i < N; i++) {
			sb.append(ans[i] + " ");
		}
		
		System.out.println(sb);
	}
}

👨🏻‍💻 결과

 

 

 

 

🔗 링크

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

17298번: 오큰수