前K个高频元素
题目链接:前K个高频元素
思路:
解法一:
- 首先对整个数组排序,复杂度O(nlgn)
- 再对每个数字出现的频度排序
- 最后找出前K个数字
解法二:
- 将数组遍历存入map集合中,value值存放出现次数
- 构造一个长度为nums.length+1的lists数组
- 将频度作为下标添加至lists[i]集合中
- 最后按标从大到小遍历,获取集合中的值,添加至结果集,加入k个即可
public List<Integer> topKFrequent(int[] nums, int k) {
Map<Integer,Integer> map = new HashMap<Integer,Integer>();
for(int num:nums) {
map.put(num,map.getOrDefault(num, 0)+1);
}
//这里加1的原因是不必单独处理下标
List[] lists = new ArrayList[nums.length+1];
for(int key:map.keySet()) {
//取出key值对应的元素个数
int temp = map.get(key);
if(lists[temp]==null) {
lists[temp] = new ArrayList<Integer>();
}
//这里就是用list存储的原因,因为同一个value值可能对应着好几个key值
lists[temp].add(key);
}
List<Integer> result = new ArrayList<Integer>();
for(int i=lists.length-1;result.size()<k&&i>=0;--i) {
if(lists[i]!=null) {
result.addAll(lists[i]);
}
}
return result;
}