欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

给定一个非空的整数数组,返回其中出现频率前 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);
    }
相关标签: