给定一个非空的整数数组,返回其中出现频率前 k 高的元素。
示例 1:
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:
输入: nums = [1], k = 1
输出: [1]
说明:
你可以假设给定的 k 总是合理的,且 1 ≤ k ≤ 数组中不相同的元素的个数。
你的算法的时间复杂度必须优于 O(n log n) , n 是数组的大小。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/top-k-frequent-elements
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
分桶法:
1.遍历数组将每个元素出现的次数存放在map中;
2.把每个元素出现的次数作为索引,对应的值则是:由每个对应的出现的次数的元素组成的list;
3.遍历桶数组,注意要从后向前遍历,这样出现的次数多的元素会先出现;
代码:
public List<Integer> topKFrequent(int[] nums, int k) {
ArrayList<Integer>[] bucket = new ArrayList[nums.length + 1];
Map<Integer,Integer> map = new HashMap<Integer, Integer>();
for (int i = 0;i < nums.length; i++) {
map.put(nums[i], map.getOrDefault(nums[i],0) + 1);
//map.getOrDefault意思就是当Map集合中有这个key时,就使用这个key值,如果没有就使用默认值
}
for (Integer key : map.keySet()) {//map.keySet()返回所有key值的列表
Integer value = map.get(key);
if (bucket[value] == null) {
bucket[value] = new ArrayList();
}
bucket[value].add(key);
}
List<Integer> res = new ArrayList<>();
for (int i = nums.length; i >=0 && res.size() < k ;i--) {
if (bucket[i] != null) {
res.addAll(bucket[i]);
}
}
return res;
}