题目描述
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
思路:要考虑边界情况,假如k>arr.size(),则要返回一个空的vector。对于k<=arr.size()的情况,则先排序,然后去排序后数组的前k个数。
代码:
class Solution {
public:
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
sort(input.begin(),input.end());
vector<int> kmin;
if(k>input.size())
{
return kmin;
}
for(int i=0;i<k;i++)
{
kmin.push_back(input[i]);
}
return kmin;
}
};