求数组中最小的k个数以及海量数据最大堆、multiset解决方案

最新推荐文章于 2025-10-01 12:00:21 发布

原创

最新推荐文章于 2025-10-01 12:00:21 发布 · 2.4k 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#海量数据 #海量数据最小k个数 #数组最小k个数 #最大堆 #multiset

【题目】

输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4,。

【方案一】

主要有两种方案。第一是利用我们熟知的 partition 算法，它是快速排序的核心，相信每个人都会。它可以用来求取数组的任意第 k 大的数，时间复杂度是O(n)。我们不断对数据 partition，当它返回的 index 为第 k-1 是，那么就说明前 k 个数（包括 index对应的数）就是最小的 k 个数了。因为 index 对应数的左侧都比它小，一共 0~k-2 即 k-1 个，加上它自己，就是 k 个了。

代码：

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        const int size = input.size();
        vector<int> res;
        if(size == 0 || k <= 0 || k > size)
            return res;
        if(k == size)
            return input;
    
        int start = 0;
        int end = size - 1;
        int index = partition(input, start, end);
        while(index != k - 1){ 
            if(index > k - 1)
                end = index - 1;
            else
                start = index + 1;
            index = partition(inp