前K个高频元素

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

示例 1:

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

输出: [1,2]

class Solution{
    public int[] topKFrequent(int[]  nums, int k){
        Map<Integer, Integer> occurrences = new HashMap<Integer, Integer>();
        for(int num:nums){
           occurrences.put(num, occurrences.getOrDefault(num, 0) + 1;
        }
        PriorityQueue<int[]> queue = new PriorityQueue<int[]>(new Comparater<int[]>()
        {
           public int Compare(int[] m, int[] n){
                return m[1] - n[1];
           }
        });
        for(Map.Entry<Integer, Integer> entry: occurrences.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;
    }
}