刷怪日记-前k个高频元素

前k个高频元素

题目

解题思路

• 抓住题目关键词，频率前k高的元素，一般统计元素数量使用哈希表
• 由于哈希表是无序的，所以我们要对哈希表的元素进行排序
• 前k个元素，此时我们可以使用大顶堆或者小顶堆来解决
• 使用大顶堆弹出前k个堆顶元素即可

具体代码

public int[] topKFrequent(int[] nums, int k){
        int[] res = new int[k];
        HashMap<Integer, Integer> map = new HashMap<>();
        // 使用哈希表
        for (int i = 0; i < nums.length; i++) {
            // 如果存在该key则直接value+1并重新put，如果没有新加入该key，并初始值为0+1
            map.put(nums[i], map.getOrDefault(nums[i], 0) + 1);
        }
        PriorityQueue<Map.Entry<Integer, Integer>> priorityQueue = new PriorityQueue<>(new Comparator<Map.Entry<Integer, Integer>>() {
            @Override
            public int compare(Map.Entry<Integer, Integer> map1, Map.Entry<Integer, Integer> map2) {
                return map2.getValue() - map1.getValue();
            }
        });
        for(Map.Entry<Integer, Integer> item : map.entrySet()){
            priorityQueue.add(item);
        }
        for (int i = 0; i < k; i++) {
            res[i] = priorityQueue.poll().getKey();
        }

        return res;
    }