自己最朴实的想法就是遍历一遍,得到值和次数的map,再对map中的次数进行堆排序,或者快排得到前k大的数。官方解答和我想的一样。
但看到了下面这个桶排序,其实还是以空间换时间的思路
class Solution {
public int[] topKFrequent(int[] nums, int k) {
int[] result=new int[k];
HashMap<Integer,Integer> map=new HashMap();
for(int num:nums){
if(map.containsKey(num)){
map.put(num,map.get(num)+1);
}else{
map.put(num,1);
}
}
//桶排序
//将频率作为数组下标,对于出现频率不同的数字集合,存入对应的数组下标
List<Integer>[] list=new List[nums.length+1];
for(int key:map.keySet()){
int i=map.get(key);
if(list[i]==null){
list[i]=new ArrayList();
}
list[i].add(key);
}
int count=0;
for(int i=list.length-1;i>=0&&count<k;i--){
if(list[i]==null)continue;
for(Integer n:list[i]){
result[count++]=n;
}
}
return result;
}
}