[백준] 17298번 오큰수 - Java, 자바

Kim Ji Eun·2022년 1월 6일
0
post-custom-banner

문제

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

코드


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

// 17298번 오큰수
public class boj_3_17298 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        Stack<Integer> stack = new Stack<>();

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

        int[] seq = new int[N];

        StringTokenizer st = new StringTokenizer(br.readLine(), " "); // 단위로 읽어드릴수 있는 라인 추가

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

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

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

        StringBuilder sb = new StringBuilder();
        for (int i = 0; i < N; i++) {
            sb.append(seq[i]).append(' ');
        }

        System.out.println(sb);
    }
}

참고
https://st-lab.tistory.com/196

profile
Back-End Developer
post-custom-banner

0개의 댓글