Leetcode 347:前K个高频元素

看到这道题一开始的思路是：使用map：键存放nums，值存放nums出现的次数。然后通过对值排序的方法找出前k个元素。这样也能做，代码如下：

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        Map<Integer,Integer> map=new HashMap<>();
        int [] res=new int[k];
        for(int i=0;i<nums.length;i++){
            map.put(nums[i],map.getOrDefault(nums[i],0)+1);
        }

        List<Map.Entry<Integer, Integer>> entries = new ArrayList<>(map.entrySet());
        Collections.sort(entries, (e1, e2) -> e2.getValue().compareTo(e1.getValue()));

        for (int i = 0; i < k && i < entries.size(); i++) {
            res[i]=entries.get(i).getKey();
        }
        return res;
    }
}

官方题解给出的是用优先级队列做，而且用的是小顶堆。大顶堆的话，每次移除的都是最大的元素(×)

代码如下：

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        int [] res=new int[k];
        Map<Integer,Integer> map=new HashMap<>();
        for(int i=0;i<nums.length;i++){
            map.put(nums[i],map.getOrDefault(nums[i],0)+1);
        }
        PriorityQueue<Integer> pq=new PriorityQueue<>(new Comparator<Integer>(){
            public int compare(Integer a,Integer b){
                return map.get(a)-map.get(b);
            }
        });
       for(Integer key:map.keySet()){
           if(pq.size()<k) pq.add(key);
           else if(map.get(key)>map.get(pq.peek())){
               pq.remove();
               pq.add(key);
           }
       }
       for(int i=0;i<k;i++){
           res[i]=pq.remove();
       }
       return res;
    }
}