Java LeetCode 347. 前 K 个高频元素

给定一个非空的整数数组，返回其中出现频率前 k 高的元素。
示例 1:
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:
输入: nums = [1], k = 1
输出: [1]

先把数组里每个元素出现的次数记录到hash中，然后通过小顶堆，构造优先级队列，把小的放到堆顶，每次和堆顶比较，大于堆顶，则把堆顶删除然后吧这个数放入。

class Solution {
    
    
    public int[] topKFrequent(int[] nums, int k) {
    
    
        Map<Integer,Integer> hash = new HashMap();

        for(int i:nums){
    
    
            if(hash.containsKey(i)){
    
    
                hash.put(i,hash.get(i)+1);
            }
            else{
    
    
                hash.put(i,1);
            }
        }

        PriorityQueue<Integer> que = new PriorityQueue(k,(v1,v2)->{
    
    
            return hash.get(v1)-hash.get(v2);
        }
            
        );

        for(int key:hash.keySet()){
    
    
            if(que.size()<k){
    
    
                que.offer(key);
            }
            else{
    
    
                if(hash.get(key)>hash.get(que.peek())){
    
    
                    que.poll();
                    que.offer(key);
                }
            }
            
        }

        int list[] = new int[k];

        for(int i=0;i<k;i++){
    
    
            list[i]=que.poll();
        }

        return list;
    }
}