思路如下:
1、首先用HashMap存储每个key及其对应的个数value
2、使用优先队列PriorityQueue,重写比较方法,把map的key从大到小进行排列。
3、使用List获取queue中的结果,Collections.reverse(list)将其倒序。
4、关键是要想到用PriorityQueue优先队列解题。
class Solution {
public List<Integer> topKFrequent(int[] nums, int k) {
//HashMap存值
Map<Integer,Integer>map=new HashMap<>();
for(int num:nums){
map.put(num,map.getOrDefault(num,0)+1);
}
//优先队列比较方法重写
PriorityQueue<Integer> queue=new PriorityQueue<>((n1,n2)->map.get(n1)-map.get(n2));
int count=1;
//把map中的key按值value的大小全部放到queue中
for(int key:map.keySet()){
queue.add(key);
}
//保证queue只有k个
while(queue.size()>k){
queue.poll();
}
//list获取queue的值
List<Integer>list=new ArrayList<>();
while(!queue.isEmpty()){
list.add(queue.poll());
}
//倒序list,方法API要记住
Collections.reverse(list);
return list;
}
}
原题地址:
347. 前 K 个高频元素