牛客网_剑指offer_最小的k个数

题目描述

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

解题思路

保持一个排序号的k个值的序列，遍历数组，当遍历的值小于序列的最后一个值的时候，就把这个数字插入进去，并且将最后的那个值弹出即可。
具体代码如下：

class Solution 
{
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) 
    {
        //维持k个数字的列表，保持这里面的数字是最小的就好
        vector<int> res{};
        if (input.empty() || k <= 0 || input.size() < k) return res;
        
        for(auto x : input)
        {
            if (res.size() < k) res.push_back(x); //可以在push的时候就保持顺序，在后面替换的时候遍历，找到比遍历值大的地方的前面insert，这样做的话时间复杂度是o(n)。
            else if (res.size() == k) 
            {
                sort(res.begin(), res.end());
                if (x < res.back())
                {
                    res.pop_back();
                    res.push_back(x);
                    sort(res.begin(), res.end());
                }
            }
        }
        
        return res;
    }
};