给定一个非空的整数数组,返回其中出现频率前 k 高的元素。
示例 1: 输入: nums = [1,1,1,2,2,3], k = 2 输出: [1,2]
示例 2: 输入: nums = [1], k = 1 输出: [1]
思路:首先应该考虑用HashMap来做,建立数字和其出现次数的映射,然后再按照出现次数进行排序。我们可以用堆排序来做,使用一个最大堆来按照映射次数从大到小排列,在java中使用PriorityQueue来实现。
public List topKFrequent(int[] nums, int k) {
//第一步:存放值 和值出现的次数
Map map=new HashMap<>();
for (int num : nums) {
if(map.get(num) == null){
map.put(num, 1);
}else{
map.put(num, map.get(num) + 1);
}
}
//第二步:生成优先队列
PriorityQueue queue = new PriorityQueue<>(new Comparator() {
@Override
public int compare(Integer a, Integer b) {
return map.get(b) - map.get(a);
}
});
for(Integer key : map.keySet()){
queue.add(key);
}
List list = new ArrayList<>();
for(int i = 0; i < k; i++){
list.add(queue.remove());
}
return list;
}