Top K Frequent Elements
给定一个非空的整数数组,返回其中出现频率前 k 高的元素。
例如,
给定数组 [1,1,1,2,2,3] , 和 k = 2,返回 [1,2]。
注意:
- 你可以假设给定的 k 总是合理的,1 ≤ k ≤ 数组中不相同的元素的个数。
- 你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。
分析
关于O(nlogn)的时间复杂度,一致都很懵,只知道优于O(n^2)差于O(n),于是乎这道题直接就放弃考虑时间复杂度了
思路是用HashMap记录<元素1,频次1>,然后将其转化为数组,数组下标记录频次,值为元素。这样我们倒序遍历k个元素,即可得到前K高的元素
代码
class Solution {
//HashMap记录<元素,频次>键值对
//用数组下标映射元素的出现次数
//两次遍历数组即可
//缺点在于需要额外的数组空间,比较耗空间
public List<Integer> topKFrequent(int[] nums, int k) {
List<Integer>[] fre = new ArrayList[nums.length+1];
List<Integer> list = new ArrayList<>();
HashMap<Integer,Integer> map = new HashMap<>();
for(int i:nums){
if(map.containsKey(i)){
map.put(i,map.get(i)+1);
}else{
map.put(i,1);
}
}
Iterator iter = map.entrySet().iterator();
while(iter.hasNext()){
Map.Entry<Integer,Integer> entry = (Map.Entry)iter.next();
if(fre[entry.getValue()]==null){
ArrayList<Integer> temp = new ArrayList<>();
temp.add(entry.getKey());
fre[entry.getValue()] = temp;
}else{
fre[entry.getValue()].add(entry.getKey());
}
}
for(int i=nums.length;i>=0&&k>list.size();i--){
if(fre[i]!=null){
list.addAll(fre[i]);
}
}
return list;
}
}