347. 前 K 个高频元素

链接：力扣

题目：

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

示例：

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

代码：

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        Map<Integer, Integer> occurences = new HashMap<Integer, Integer>();
        //哈希表存储num的各个数的个数
        for(int num:nums){
            occurences.put(num, occurences.getOrDefault(num, 0) + 1);
        }

        //基于优先级的无界优先级队列
        // int[] 的第一个元素代表数组的值，第二个元素代表了该值出现的次数
        PriorityQueue<int[]> queue = new PriorityQueue<int[]>(new Comparator<int[]>() {
            public int compare(int[] m, int[] n) {
                return m[1] - n[1];
            }
        });

        for(Map.Entry<Integer, Integer> entry : occurences.entrySet()){
            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[] ret = new int[k];
        for (int i = 0; i < k; ++i) {
            ret[i] = queue.poll()[0];
        }
        return ret;


    }
}