LeetCode #347. 상위 K 빈도 요소

Source: Dev.to

복잡도 분석

• 시간 복잡도: O(n log n) (정렬에 의해 지배됨)
• 공간 복잡도: O(n) HashMap 및 List에 대해

솔루션

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        Map map = new HashMap<>();
        for (int num : nums) {
            map.put(num, map.getOrDefault(num, 0) + 1);
        }

        List list = new ArrayList<>(map.keySet());
        list.sort((a, b) -> Integer.compare(map.get(b), map.get(a)));

        int[] result = new int[k];
        for (int i = 0; i < k; i++) {
            result[i] = list.get(i);
        }
        return result;
    }
}