题目来源:
牛客网
编程链接
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
先简单的把数组排序,求前K个即可,考虑输入数组个数小于K的情况
class Solution {
public:
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
vector<int> ret;
if(input.size()<k)
return ret;
sort(input.begin(),input.end());
for(int i=0;i<k;ret.push_back(input[i++]));
return ret;
}
};