[백준] 17298 오큰수 / 자바(Java)

문제

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

17298번: 오큰수

 

---

해설

오큰수는 수열에서 특정 숫자보다 오른쪽에 있는 수 중 큰 가장 가까운 수이다.

들어오는 값의 순서를 유지하고 오큰수가 정해진 수에 관여하지 않기 위해 stack을 사용한다.

 

앞에 있는 값부터 순서대로 stack에 넣으면 stack에는 값이 역순으로 들어가게 됩니다. 그러다 오큰수에 해당하는 stack의 peek값보다 큰 수가 들어온다면 stack에서 그 값을 pop해주고 오큰수로 배열에 저장해 주며 pop과 저장을 가능할 때 까지 반복해 주고 pop되지 않고 남는 값들은 오큰수가 없는 것이므로 -1을 넣어 간단하게 해결 가능하다.

 

stack의 넣은 것의 역순으로 나오는 성질을 사용해 간단히 해결하여 올바른 자료구조가 문제를 쉽게 만들어 줄 수 있음을 알았다.

주의점

• 수열 A의 길이가 1,000,000으로 길어 일일히 비교하는 방식은 시간이 초과된다.

---

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Deque;
import java.util.LinkedList;
import java.util.Stack;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));

        int N = Integer.parseInt(br.readLine());

        int[] arr = new int[N];

        StringTokenizer st = new StringTokenizer(br.readLine());
        for (int i = 0; i < N; i++) {
            arr[i] = Integer.parseInt(st.nextToken());
        }
        Stack<Integer> stack = new Stack<>();
        for (int i = 0; i < N; i++) {
            while (!stack.isEmpty() && arr[i]>arr[stack.peek()]){
                arr[stack.pop()]=arr[i];
            }
            stack.push(i);
        }

        while(!stack.isEmpty()){
            arr[stack.pop()] = -1;
        }

        StringBuilder sb = new StringBuilder();

        for (int i = 0; i < N; i++) {
            sb.append(arr[i]).append(" ");
        }

        System.out.println(sb);
    }
}