给定一个非空的整数数组,返回其中出现频率前 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 是数组的大小。
思路:
- 利用哈希映射以<number, frequency>的形式存储每个数的频率;
- 用桶排序的思想将频率为frequency的数放入桶中(频率最大为nums.length),下标为对应数的频率。考虑到有的数频率可能相同,桶中的元素应该为一个数组。
- 由于找频率前K高的元素,倒序遍历桶记录答案。
代码如下:
class Solution {
public List<Integer> topKFrequent(int[] nums, int k) {
List<Integer> ans = new ArrayList<Integer>(); //记录答案
Map<Integer, Integer> map = new HashMap<Integer, Integer>();
for(int x : nums){
map.put(x, map.getOrDefault(x, 0) + 1);
} //记录每个数的频率
List<Integer>[] bucket = new List[nums.length + 1];
for(int key : map.keySet()){
int frequency = map.get(key);
if(bucket[frequency] == null) bucket[frequency] = new ArrayList<Integer>();
bucket[frequency].add(key);
} //用桶装频率对应的所有数
for(int i=bucket.length-1; i>=0 && ans.size() < k; i--){
if(bucket[i] != null){
ans.addAll(bucket[i]);
}
}
return ans;
}
}
总结:
- map.getOrDefault(Object key, Object defaultValue):存在key返回对应value,否则返回默认的defaultValue。
- map.keySet():返回map中关键字所组成的集合。
- 数据结构对象运用之前要判断是否为空,如if(bucket[frequency] == null) bucket[frequency] = new ArrayList();。