题目描述
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
代码实现
class Solution {
public:
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
/*超时!!!
vector<int> arr;
if(input.size()==0||k==0||k>input.size()){
return arr;
}
int size = input.size();
for(int i =0;i<k;i++){
int min = input[i];
for(int j = 1;j<size;k++){
if(min> input[j]){
min = input[j];
input[j] = input[i];
input[i] = min;
}
}
int ii = input[i];
arr.push_back(ii);
}
return arr;
*/
//利用系统函数sort(类似快排,时间复杂度O(n*log n))
vector<int> arr;
if(input.size()== 0||k==0||input.size()<k){
return arr;
}
sort(input.begin(),input.end());
for(int i =0;i<k;i++){
arr.push_back(input[i]);
}
return arr;
}
};