boj)17299 - 오등큰수

2020. 9. 15. 17:02PS/boj

import java.io.*;
import java.util.Stack;
import java.util.StringTokenizer;

// #17299_오등큰수
public class boj_17299 {
    public static void main(String[] args) throws Exception {
        Stack<Integer> s = new Stack<>();
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        int n = Integer.parseInt(br.readLine());
        int[] arr = new int[n];
        int[] ngf = new int[n];
        int[] F = new int[1000001];

        StringTokenizer st = new StringTokenizer(br.readLine());
        for (int i = 0; i < n; i++) {
            arr[i] = Integer.parseInt(st.nextToken());
            F[arr[i]]++;
        }

        for (int i = 0; i < n; i++) {
            if (s.isEmpty()) {
                s.push(i);
            }

            while (!s.isEmpty() && F[arr[s.peek()]] < F[arr[i]]) {
                ngf[s.pop()] = arr[i];
            }

            s.push(i);
        }

        while (!s.isEmpty()) {
            ngf[s.pop()] = -1;
        }

        for (int i = 0; i < n; i++) {
            bw.write(ngf[i] + " ");
        }
        bw.flush();
        bw.close();
    }
}

- 스택 사용

- 오큰수와 비슷한 개념인데 횟수조건이 추가됨, 따로 F 배열을 만들어서 수의 출현횟수를 카운팅

- 런타임오류에 계속 걸렸는데 F배열의 크기를 n으로 줬었음, 수의 범위는 1,000,000 까지니까 배열 크기를 바꾸니 해결

'PS > boj' 카테고리의 다른 글

boj)6588 - 골드바흐의 추측  (0) 2020.09.15
boj)1929 - 소수 구하기  (0) 2020.09.15
boj)17298 - 오큰수  (0) 2020.09.15
boj)10799 - 쇠막대기  (0) 2020.09.14
boj)17413 - 단어 뒤집기 2  (0) 2020.09.14