[백준] 오큰수 17298 - java

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

17298번: 오큰수

public class 오큰수 {
	public static void main(String[] args) throws NumberFormatException, IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int N = Integer.parseInt(br.readLine());
		Stack<int []> stack = new Stack<>();
		StringTokenizer st = new StringTokenizer(br.readLine());
		int answer[] = new int[N];
		stack.push(new int[] {Integer.parseInt(st.nextToken()), 0});
		StringBuilder sb = new StringBuilder();
		for(int i=1; i<N; i++) {
			int cur = Integer.parseInt(st.nextToken());
			while(!stack.isEmpty()){
				if(stack.peek()[0] < cur) {
					answer[stack.pop()[1]] = cur;
				}
				else {
					break;
				}
			}
			stack.push(new int[] {cur, i});
		}
		while(!stack.isEmpty()) {
			answer[stack.pop()[1]] = -1;
		}
		
		for(int num : answer) {
			sb.append(num).append(" ");
		}
		
		System.out.println(sb.toString());
	}
}

스택에는 수열의 값과 인덱스 값을 받을 수 있게 선언했습니다.

1. 처음 값을 스택에 넣는다.

2. 스택의 peek 값이 현재값보다 작다면 pop을 하고 answer배열에 해당 인덱스에 현재값을 할당합니다.

3. peek값이 현재값보다 클때까지 2를 반복합니다.

4. 스택에 현재값을 push합니다.

5. 스택에 남아있는 값들은 answer에 -1로 할당합니다.