[JAVA][Baekjoon] 17298번 오큰수 ⭐️⭐️⭐️⭐️⭐️

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

 

📌 접근 방식

• 반복문을 이용해 오큰수를 찾는다면 시간 초과가 발생
• 스택에 새로 들어오는 수가 top에 존재하는 수보다 크면 그 수는 오큰수!

 

✅ PASS CODE

import java.util.*;
import java.io.*;

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 []A = new int[n];
        int []ans = new int [n];

        String[] str = br.readLine().split(" ");
        for (int i = 0; i < n; i++) {
            A[i] = Integer.parseInt(str[i]);
        }
        Stack<Integer> stack = new Stack<>();
        stack.push(0);
        for(int i = 1; i < n; i++) {
            while (!stack.empty() && A[i] > A[stack.peek()]){
                ans[stack.pop()] = A[i];
            }
            stack.push(i);
        }
        while(!stack.empty()){
            ans[stack.pop()] = -1;
        }
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        for(int i = 0; i < n; i++) {
            bw.write(ans[i] + " ");
        }
        bw.write("\n");
        bw.flush();

    }
}

• System.out / Scanner로 입출력을 진행하였는데 시간초과가 발생했다
  ➡️ BufferedWriter / BufferedReader 사용!!
  ❗️flush 해줘야함

 

728x90