347. 前 K 个高频元素

题目描述

给你一个整数数组 nums 和一个整数 k ，请你返回其中出现频率前 k 高的元素。你可以按 任意顺序 返回答案。

示例 1:

输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:

输入: nums = [1], k = 1
输出: [1]

提示：

1 <= nums.length <= 105
k 的取值范围是 [1, 数组中不相同的元素的个数]
题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的

---

做题思路

• 想要把本题做出来，必须要对PriorityQueue优先队列有一定的了解，其内部本质是一个堆，进行堆排序。
• 如果不指定排序规则，对于普通类型来说，默认按照升序排列，而对于自定义类型的引用变量，我们可以通过 Comparator比较器来自定义比较规则，对于本题我们就需要使用自定义比较器指定按照链表结点的值的升序排列
• 知道这个优先队列我们就可以容易的做这个题目了
• 但是对于本题来说，我们还需要自定义一个内部类，用来保存每个元素对应的数量，同时我们需要先通过Map集合将每个元素对应的数量保存起来，然后通过遍历Map集合，根据题目中给出的找出前k个高频元素来作为判断条件

代码实现

class Solution {
    class numC{
        int num;
        int count;
        public numC(int num,int count){
            this.num=num;
            this.count=count;
        }
    }
    public int[] topKFrequent(int[] nums, int k) {
        if(k>nums.length)return new int[0];
        Queue<numC> queue=new PriorityQueue<>(new Comparator<numC>(){
            public int compare(numC a,numC b){
                return a.count-b.count;
            }
        });

        Map<Integer,Integer> map=new HashMap<>();
        for(int num:nums){
            map.put(num,map.getOrDefault(num,0)+1);
        }

        for(Map.Entry<Integer,Integer> entry:map.entrySet()){
            int count=entry.getValue();
            int num=entry.getKey();
            if(queue.size()<k){
                queue.offer(new numC(num,count));
            }else{
                numC n=queue.peek();
                if(n.count<count){
                    queue.poll();
                    queue.offer(new numC(num,count));
                }
            }
        }

        int[] res=new int[k];
        for(int i=0;i<k;i++){
            res[i]=queue.poll().num;
        }
        return res;
    }
}

题目链接

347. 前 K 个高频元素