75.前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 <= 10⁵
• k 的取值范围是 [1, 数组中不相同的元素的个数]
• 题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的

代码：

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        //使用哈希表统计每个数字出现的频率
        Map<Integer,Integer>mp = new HashMap<>();
        //遍历nums数组统计出现频率，如果数字已存在哈希表则计数加一，否则初始化为1
        for(int i = 0;i<nums.length;i++)mp.put(nums[i],mp.getOrDefault(nums[i],0)+1);
        //创建最小堆，堆中存储的是int数组，数组的第一个元素是数字，第二个元素是出现次数
        //比较器按出现次数升序排列，堆顶是当前堆出现次数最小的元素
        PriorityQueue<int[]>queue = new PriorityQueue<>(new Comparator<int[]>(){
            public int compare(int[] m,int[] n){
                //按出现次数比较
                return m[1] - n[1];
            }
        });
        //遍历频率哈希表，维护大小为k的最小堆
        for(Map.Entry<Integer,Integer>entry:mp.entrySet()){
            //num为当前数字，count为当前数字的出现次数
            int num = entry.getKey(),count = entry.getValue();
            //若堆已满，只有当当前数字出现次数大于堆顶元素时才替换
            if(queue.size() == k){
                if(queue.peek()[1]<count){
                    //移除堆顶元素（出现次数最少的元素）
                    queue.poll();
                    //加入当前元素
                    queue.offer(new int[]{num,count});
                } 
            }else queue.offer(new int[]{num, count});//堆未满，直接加入
        }
        //将堆中的元素转为结果数组
        int[] res = new int[k];
        //从堆中取出元素，注意这里取出的是出现频率从低到高的顺序
        //由于可以按任意顺序返回答案，所以直接按照取出顺序即可
        //只需要数字部分，不需要出现次数
        for(int i=0;i<k;i++)res[i] = queue.poll()[0];
        //返回结果
        return res;
    }
}