[백준] 17299 오등큰수 - 자바

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

17299번: 오등큰수

문제 풀이

오큰수 문제에서 약간 변형된 문제로 이 문제는 값을 비교하지 않고 값의 등장횟수를 비교해 준다.

오큰수 문제의 해결법은 여기를 참고하길 바란다.

따라서 오등큰수를 구하는 로직은 같으며 비교를 위한 등장횟수만 hashMap을 통해 관리해주면 된다.

코드

package 백준;

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

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

        HashMap<Integer, Integer> numCnt = new HashMap<>();
        Stack<Integer> s = new Stack<>();

        int n = Integer.parseInt(br.readLine());
        int[] result = new int[n];
        int[] nums = Arrays.stream(br.readLine().split(" ")).mapToInt(Integer::parseInt).toArray();
        for (int num : nums) {
            numCnt.put(num, numCnt.getOrDefault(num, 0) + 1);
        }

        for (int i = 0; i < n; i++) {
            // 순차탐색하고 있는 값의 등장횟수가 스택에 담긴 idx의 값의 등장횟수보다 크다면
            while (!s.isEmpty() && numCnt.get(nums[s.peek()]) < numCnt.get(nums[i])) {
                result[s.pop()] = nums[i];
            }
            s.push(i);
        }

        // 스택에 남아있다는 것은 오등큰수에 해당 X -> 전부 -1로
        while (!s.isEmpty()) {
            result[s.pop()] = -1;
        }

        Arrays.stream(result).forEach(x -> sb.append(x + " "));
        System.out.println(sb);
    }
}