[Leetcode] 347. 前 K 个高频元素

[Leetcode] 347. 前 K 个高频元素

题目链接：347. 前 K 个高频元素

思路：

• 方法一
  官方给出的做法采用小顶堆的方法，维护堆中的k个数，当堆中个数小于k个的时候，直接add就可以了，当堆中个数大于k的时候，每次poll一个，add一个，这样每次poll的都是最小的数，这样到最后就维护了k个最大的。
• 方法二
  我们也可以采用Java中大顶堆的方法，将所有的数字和数字出现的次数都add进大顶堆中，即优先队列。poll出k次即可。

代码（Java）：

// 大顶堆的做法
class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        if(nums == null || k <= 0) return null;
        Map<Integer, Integer> map = new HashMap<>(); // 统计每个数字出现的次数
        // 构造一个大顶堆，对于二元组(num,cnt)，表示数字和次数，我们对次数进行排序
        PriorityQueue<int[]> queue = new PriorityQueue<>( (pair1, pair2)->(pair2[1]-pair1[1]) );
        int[] res = new int[k]; // 结果数组
        int index = 0;

        // 统计数字出现的次数
        for (int num : nums) {
            map.put(num, map.getOrDefault(num,0) + 1);
        }
        // 将二元组插入到大顶堆
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            queue.add(new int[]{entry.getKey(), entry.getValue()});
        }

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