题目描述
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
解答
方法一
时间复杂度O(NlognN)
class Solution {
public:
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
vector<int> res;
if(input.size()<k) return res;
sort(input.begin(),input.end());
for(int i=0;i<k;++i)
res.push_back(input[i]);
return res;
}
};
方法二
时间复杂度O(NlognK)
class Solution {
public:
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
vector<int> res;
if(input.size()<k) return res;
for(int i=k;i<input.size();++i)
{
for(int j=0;j<k;++j)
{
if(input[j]>input[i])
{
int temp=input[j];
input[j]=input[i];
input[i]=temp;
}
}
}
for(int i=0;i<k;++i)
{
res.push_back(input[i]);
}
return res;
}
};