题目:给定一个非空的整数数组,返回其中出现频率前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;
}
}