给定一个非空的整数数组,返回其中出现频率前 k 高的元素。要求时间复杂度优于O(nlogn)
程序员文章站
2024-03-15 22:47:54
...
给定一个非空的整数数组,返回其中出现频率前 k 高的元素。要求时间复杂度优于O(nlogn)
示例 1:
输入: nums = [1,1,1,2,2,3], k = 2
输出: [1,2]
示例 2:
输入: nums = [1], k = 1
输出: [1]
说明:
你可以假设给定的 k 总是合理的,且 1 ≤ k ≤ 数组中不相同的元素的个数。
桶排序
public List<Integer> topKFrequent(int[] nums, int k) {
HashMap<Integer, Integer> map = new HashMap<>();
for(int i : nums) { // 统计频率,key为元素,value为次数
map.put(i, map.getOrDefault(i, 0) + 1);
}
// 创建nums.length + 1个桶
List<Integer>[] bucket = new List[nums.length + 1];
// 遍历map,根据value值放到对应的桶中
for(Map.E***y<Integer, Integer> e : map.e***ySet()) {
Integer value = e.getValue();
if(bucket[value] == null) {
bucket[value] = new ArrayList<Integer>();
}
bucket[value].add(e.getKey());
}
List<Integer> freList = new ArrayList<>();
// 桶的编号表示出现次数,所以倒数桶
for(int j = bucket.length - 1; j > -1 && freList.size() < k; j--) {
if(bucket[j] != null)
freList.addAll(bucket[j]);
}
return freList;
}
==================================
下面自己练习一下集合
public static void main(String[] args) {
int[] arr = {1,1,1,1,1,1,2,2,2,2,2,3};
int k = 2;
getResult(arr,k);
}
private static void getResult(int[] arr, int k) {
HashMap<Integer,Integer> map = new HashMap<>();
List<Integer>[] listArr = new List[arr.length+1];
for (int i=0;i<arr.length;i++){
map.put(arr[i],map.getOrDefault(arr[i],0)+1);
}
for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
if (listArr[entry.getValue()]==null){
listArr[entry.getValue()]=new ArrayList<>();
}
listArr[entry.getValue()].add(entry.getKey());
}
List<Integer> ls = new ArrayList<>();
for (int i=listArr.length-1;i>=0&&ls.size()<k;i--){
if (listArr[i]!=null){
ls.addAll(listArr[i]);
}
}
System.out.println(ls);
}
=====================================
public static void main(String[] args) {
int[] arr = {1,1,1,2,2,2,2,2,3};
// HashMap<Integer,Integer> hashMap = new HashMap<>();
// hashMap.put(1,123);
// System.out.println(hashMap.get(1));
getResult(arr);
}
private static void getResult(int[] arr) {
HashMap<Integer,Integer> map = new HashMap<>();
HashSet<Integer> set = new HashSet<>();
for (int i=0;i<arr.length;i++){
set.add(arr[i]);
}
Iterator<Integer> iterator = set.iterator();
while (iterator.hasNext()){
Integer temp = iterator.next();
for (int j=0;j<arr.length;j++){
if (temp==arr[j]){
if (map.get(temp)==null){
map.put(temp,1);
}else {
// Integer lastValue = map.get(temp);
// map.put(temp,++lastValue);
map.put(temp,map.get(temp)+1);
}
}
}
}
System.out.println(map.toString());
Set<Map.Entry<Integer, Integer>> entrySet = map.entrySet();
ArrayList<Map.Entry<Integer, Integer>> list = new ArrayList<>(entrySet);
System.out.println(list);
Collections.sort(list, new Comparator<Map.Entry<Integer, Integer>>() {
@Override
public int compare(Map.Entry<Integer, Integer> o1, Map.Entry<Integer, Integer> o2) {
return o2.getValue() - o1.getValue();
}
});
System.out.println(list);
List<Integer> ls = new ArrayList<>();
for (int i=0;i<2;i++){
ls.add(list.get(i).getKey());
}
System.out.println(ls);
}