天天看點

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允許