牛客网链接:最小的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;
}
};