백준

백준 17298 - Java

으엉어엉 2024. 9. 17. 17:27
728x90

 

 

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Stack;

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

        // 첫 번째 줄에서 N을 입력받음
        int N = Integer.parseInt(br.readLine());

        int[] arr = new int[N];
        int[] nge = new int[N];  // 오큰수를 저장할 배열
        Stack<Integer> stack = new Stack<>();

        String[] input = br.readLine().split(" ");

        for (int i = 0; i < N; i++) {
            arr[i] = Integer.parseInt(input[i]);
        }

        // 스택을 이용해 오큰수 구하기
        for (int i = 0; i < N; i++) {
            while (!stack.isEmpty() && arr[stack.peek()] < arr[i]) {
                nge[stack.pop()] = arr[i];
            }
            stack.push(i);
        }

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

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

        System.out.println(sb.toString().trim());
    }
}

 

Stack과 Queue 사이에서 고민하였지만 Stack이 조금 더 빨리 생각나서 Stack으로 풀어봤다. 나중에 시간이 되면 Queue나 Deque로 한번 풀어봐야겠다.

728x90

'백준' 카테고리의 다른 글

백준 1935 - Java  (0) 2024.09.19
백준 17299 - Java  (0) 2024.09.19
백준 10799 - Java  (0) 2024.09.15
백준 17413 - Java  (0) 2024.09.15
백준 11729 - Java  (1) 2024.09.14