牛客题霸–最小的K个数
做法:排序
时间复杂度:O(nlogn)
思路
这题所要求输入n个整数,找出其中最小的K个数.
注意要保证input数组大小 <= k
代码
class Solution {
public:
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
sort(input.begin(), input.end());
int sz = input.size(), cnt=0;
vector<int> ans;
for(int i=0; i<sz && cnt<k && k <= sz; i++, cnt++) {
ans.push_back(input[i]);
}
return ans;
}
};