经典题:前k个高频元素 必须要会

题目:给定一个非空的整数数组,返回其中出现频率前K高的元素
思路:建立map用来存放每个数出现的频率,这一步大家都能想到,关键是怎么从map中读出前k个高频元素;这个时候我们应该想到使用优先队列;这边有两种方式,一种是使用默认的最小堆然后设定堆的大小为k;另一种方式是将最小堆转成最大堆,直接输出前k个元素。
这里需要注意的是,队列中存入的map的entry,比较的时候直接根据entry中的getValue来比较频率。
回顾:
KeySet(): get() 遍历取出key的值
entrySet(): getKey(),getValue()
map.get(key) :获取key所对应的value
getOrDefault(key,default value) :

class Solution{
public  int[] topKFrequent(int[] nums,int k){
        HashMap<Integer,Integer> map = new HashMap<>();
        int[] res = new int[k];
        PriorityQueue<Map.Entry<Integer,Integer>> queue = new PriorityQueue<>((a,b)->(b.getValue()-a.getValue()));//按照大到小,默认是a-b,从小到大
        for(int i=0;i<nums.length;i++){
            map.put(nums[i],map.getOrDefault(nums[i],0)+1);
        }
        for(Map.Entry entry :map.entrySet()){
            queue.add(entry);//把整个entry放入到queue中
        }
        for(int j=0;j<k;j++){
            res[j] = queue.poll().getKey();
        }
        return res;
    }
}

你可能感兴趣的:(数据结构与算法)