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

题目描述

输入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;
    }
};

运行时间：2ms

占用内存：376k