代码随想录---347. 前 K 个高频元素

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

示例 1:

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

示例 2:

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

解析：

这道题目主要涉及到如下三块内容：

1. 要统计元素出现频率
2. 对频率排序
3. 找出前K个高频元素

首先统计元素出现的频率，这

一类的问题可以使用map来进行统计。

然后是对频率进行排序，这里我们可以使用一种 容器适配器就是优先级队列。

 //出现频率最多的前k个
    public int[] topKFrequent(int[] nums,int k){

        HashMap<Integer, Integer> hashMap = new HashMap<>();
        //1. 先使用hashmap
        for (int i=0;i<nums.length;i++){
            //key:nums[i] value：出现频率：
            hashMap.put(nums[i],hashMap.getOrDefault(nums[i],0)+1);
        }
        
        // 2.使用优先级队列；存入的是一个数组 ；需要自定义比较器：按元素出现频率升序的方式排列
        PriorityQueue<int[]> queue = new PriorityQueue<>((num1, num2) -> num1[1] - num2[1]);

        //遍历hashmap ---->一个键值对实体：
        for (Map.Entry<Integer, Integer> entry: hashMap.entrySet()) {
            if (queue.size()<k){
                //如果队列中没有放入值
                queue.add(new int[]{entry.getKey(),entry.getValue()});
            }else{
                //如果队列中已经放入了k个元素；

                //如果放入的值大于队头中元素出现的次数 ,就要进行比较将较小的进行出队
                if (entry.getValue()>queue.peek()[1]){
                    queue.poll();//出栈；
                    //加入新的值：
                    queue.add(new int[]{entry.getKey(), entry.getValue()});
                }
            }
        }
        //3.倒序插入前k个数
        int[] res = new int[k];

        //出现频率越多的数，越先优先出现数组的前面，所以要倒序放置元素；使出现次数少的元素排在后面
        for (int i=res.length-1;i>=0;i--){
            res[i]=queue.poll()[0];
        }
        return res;
    }