[Java][백준 17298] 오큰수

Dawon Seo·2024년 3월 20일
0

알고리즘 공부

목록 보기
7/8
post-thumbnail

1. 문제

📌 문제 보러 가기!

2. 접근법

다음의 프로세스를 따른다.

  1. 0~N-1의 인덱스 값을 스택에 순서대로 넣는다.
  2. stack의 top에 있는 값이 현재 추가하려는 값보다 크면 pop하여 꺼낸다. 이를 조건이 만족하는동안 반복한다.
  3. 현재 인덱스를 push한다.

3. 코드

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

public class BaekJoon17298 {
    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];
        int[] ansArr = new int[N];
        StringTokenizer st = new StringTokenizer(br.readLine());
        Stack<Integer> stack = new Stack<>();

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

        for (int i = 0; i < N; i++) {
            while (!stack.isEmpty() && arr[i] > arr[stack.peek()]) {
                ansArr[stack.pop()] = arr[i];
            }
            stack.push(i);
        }

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

        StringBuilder sb = new StringBuilder();

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

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

0개의 댓글