最小的k个数 — C++实现

Jeaten

于 2020-09-22 15:49:00 发布

阅读量879

点赞数

CC 4.0 BY-SA版权

分类专栏：剑指Offer 文章标签： c++ 最小k个数剑指Offer 牛客网

本文链接：https://blog.csdn.net/Jeaten/article/details/108723758

剑指Offer 专栏收录该内容

38 篇文章

订阅专栏

本文介绍了一种高效算法，用于从输入的整数数组中找到最小的K个数。通过使用固定大小为K的容器，迭代过程中保持容器内始终是最小的K个数，最终返回这些数。该算法避免了对整个数组进行排序，从而降低了时间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

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

牛客网题目链接

解题思路

当然，直接拿来排序，然后输出前k个是直接能想到的解法，但这样操作的复杂度较高。推荐一种方法：

先申请一个容量为k的容器，最开始拿数组中的值填进去，然后遍历数组，如果遍历到的值比容器里的最大值小，则将容器里最大的值替换掉，这样当数组遍历完成时变得到了最小的k个数

Note：输出前k个，则必须要保证输入的数组中有k个元素

代码实现

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector <int> res;
        if(!input.size()||k<=0||input.size()<k)return res;
        int i,loc;
        vector <int>::iterator maxv;
        for(i=0;i<k;i++)res.push_back(input[i]);
        for(i=k;i<input.size();i++){
            maxv=max_element(res.begin(),res.end());
            loc=distance(res.begin(),maxv);
            if(input[i]<*maxv){
                res.erase(res.begin()+loc);
                res.push_back(input[i]);
            }
        }
        return res;
    }
};