LeetCode 347 前K个高频元素

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        HashMap<Integer,Integer> map = new HashMap<>();
        PriorityQueue<Map.Entry<Integer,Integer>> queue = new PriorityQueue<>(new Comparator<Map.Entry<Integer,Integer>>(){
            @Override
            public int compare(Map.Entry<Integer,Integer> o1,Map.Entry<Integer,Integer> o2){
                return o2.getValue() - o1.getValue();
            }
        });
        int[] res = new int[k];
        for(int i=0;i<nums.length;i++){
            map.put(nums[i],map.getOrDefault(nums[i],0)+1);
        }
        Set<Map.Entry<Integer,Integer>> set=map.entrySet();
        for(Map.Entry<Integer,Integer> s:set){
            queue.add(s);
        }
        int j=0;
        for(int i=0;i<k;i++){
            res[j++]=queue.poll().getKey();
        }
        return res;
    }
}
           

需要注意的地方：

1.注意这个Map.Entry<>，没有构造方法，是个接口

2.对于这道题而言，频率正是意味着一种优先级，对于优先级队列而言，正合适，但是怎么更好使用呢？可以不断更改每个元素的优先级呢?元素就是Key，优先级大小就是Value

3.因为Map.Entty不能直接构造，只能通过Map来put，所以先形成HashMap，在一个一个加入队列

4.HashMap的entrySet可以用来得到Map.Entty的集合

5.如果只获得Map的value集合，排序后，无法得到对应的Key，Key不允许重复，Value允许