力扣刷题347-前k个高频元素

题目
给定一个非空的整数数组，返回其中出现频率前 k 高的元素。

题解
先用map统计数组出现频率，然后构造一个小顶堆，当小顶堆的元素个数小于k时，就一直加入小顶堆。当小顶堆元素个数等于k时，若堆顶最小元素的频率也大于即将加入的，则舍弃该数。否则弹出栈顶元素，加入该数。代码如下

public int[] topKFrequent(int[] nums, int k) {
        int n = nums.length;
        if (nums == null || n == 0 || k < 1 || k > n) return new int[0];
        int[] res = new int[k];
        Map<Integer, Integer> map = new HashMap<>();
        for (int num : nums) {
            map.put(num, map.getOrDefault(num, 0) + 1);
        }
        PriorityQueue<int[]> pq = new PriorityQueue<>(new Comparator<int[]>() {
            public int compare(int[] m, int[] n) {
                return m[1] - n [1];
            }
        });
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            int num = entry.getKey();
            int count = entry.getValue();
            if (pq.size() == k) {
                if (pq.peek()[1] < count) {
                    pq.poll();
                    pq.offer(new int[]{num, count});
                }
            } else {
                pq.offer(new int[]{num, count});
            }
        }
        for (int i = 0; i < k; i++) {
            res[i] = pq.poll()[0];
        }
        return res;
    }