解题思路如下:
1、判断数组的大小;
2、将数组元素进行排序;
3、当数组元素个数不小于K时,将数组的前K个元素取出放入另一个数组中
4、返回另一个数组,则得到前K个小的元素;代码如下:
class Solution {
public:
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
vector<int>out;
vector<int>in;
vector<int>::iterator it;
in=input;
int n;
n=in.size();
sort(in.begin(),in.end());
if(n>=k)
{
for(it=in.begin();it!=in.begin()+k;it++)
{
out.push_back(*it);
}
}
return out;
}
};