剑指offer 最小的K个数

牛客网链接：最小的K个数

题目描述

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

思路

先给vector排序，然后截取前k个元素并返回，注意处理边界。

AC代码

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> res(0);
        if(input.size()==0 || k <= 0 || k>input.size() )
            return res;
        std::sort(input.begin(), input.end());
        res.assign(input.begin(), input.begin() + k);
        return res;
    }
};