TopK

最新推荐文章于 2024-07-18 16:38:07 发布

借点头发吧

最新推荐文章于 2024-07-18 16:38:07 发布

阅读量137

点赞数

分类专栏：数据结构算法练习

本文链接：https://blog.csdn.net/weixin_47866171/article/details/108458698

版权

数据结构算法练习专栏收录该内容

27 篇文章 0 订阅

订阅专栏

本文详细解析了TopK问题的多种高效解决方案，包括快速排序变形、堆和二叉搜索树等算法的应用，以及如何在数据范围有限的情况下使用计数排序。通过具体实例，如求解前K个高频元素，展示了不同算法的实现方式和时间复杂度。

摘要由CSDN通过智能技术生成

TopK问题，不管是求前K大/前K小/第K大/第K小等，有以下方法：

O(N)：用快排变形最高效解决TopK问题
O(NlogK)：大根堆（前K小）/小根堆（前K大）
O(NlogK)：二叉搜索树
O(N): 对于数据范围有限的情况，可以直接计数排序O(N)高效解决

347. 前 K 个高频元素

给定一个非空的整数数组，返回其中出现频率前 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 是数组的大小。
题目数据保证答案唯一，换句话说，数组中前 k 个高频元素的集合是唯一的。
你可以按任意顺序返回答案。
题解：
1）Hashmap
用map记录元素出现次数，找出现次数最多的元素出现的次数，按照出现次数大小添加到结果数组

2)堆
同上记录出现次数，形成一个出现次数数组，找其前k大。建立小顶堆去遍历。如果堆的元素个数小于 k，就可以直接插入堆中。
如果堆的元素个数等于 k，则检查堆顶与当前出现次数的大小。如果堆顶更大，说明至少有 k 个数字的出现次数比当前值大，故舍弃当前值；否则，就弹出堆顶，并将当前值插入堆中。遍历完成后，堆中的元素就代表了「出现次数数组」中前 k 大的值

solution：
1）

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        int res[]=new int[k];
        Map<Integer,Integer> map=new HashMap<>();
        for(int i=0;i<nums.length;i++){
            if(map.containsKey(nums[i])) map.put(nums[i],map.get(nums[i])+1);
            else map.put(nums[i],1);
        }

        int maxTimes=0;
         for(Map.Entry<Integer, Integer> entry : map.entrySet()){
            if(entry.getValue() > maxTimes){
                maxTimes = entry.getValue();
            }
        }
        while(k > 0){
            for(Map.Entry<Integer, Integer> entry : map.entrySet()){
                if(entry.getValue() == maxTimes){
                    res[k - 1] = entry.getKey();
                    k--;
                }
            }
            maxTimes--;
        }

        return res;
    }
}

2）

class Solution {
    public int[] topKFrequent(int[] nums, int k) {
        Map<Integer, Integer> occurrences = new HashMap<Integer, Integer>();
        for (int num : nums) {
            occurrences.put(num, occurrences.getOrDefault(num, 0) + 1);
        }

        // int[] 的第一个元素代表数组的值，第二个元素代表了该值出现的次数
        PriorityQueue<int[]> queue = new PriorityQueue<int[]>(new Comparator<int[]>() {
            public int compare(int[] m, int[] n) {
                return m[1] - n[1];
            }
        });
        for (Map.Entry<Integer, Integer> entry : occurrences.entrySet()) {
            int num = entry.getKey(), count = entry.getValue();
            if (queue.size() == k) {
                if (queue.peek()[1] < count) {
                    queue.poll();
                    queue.offer(new int[]{num, count});
                }
            } else {
                queue.offer(new int[]{num, count});
            }
        }
        int[] ret = new int[k];
        for (int i = 0; i < k; ++i) {
            ret[i] = queue.poll()[0];
        }
        return ret;
    }
}

Map是java中的接口，Map.Entry是Map的一个内部接口。

Map提供了一些常用方法，如keySet()、entrySet()等方法。

keySet()方法返回值是Map中key值的集合；entrySet()的返回值也是返回一个Set集合，此集合的类型为Map.Entry。

Map.Entry是Map声明的一个内部接口，此接口为泛型，定义为Entry<K,V>。它表示Map中的一个实体（一个key-value对）。接口中有getKey(),getValue等方法