輸入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) {
int len = input.size();
vector<int> arr;
for(int i = len-1;i>0;i--)
{
bool flag = true;
for(int j=0;j<i;j++)
{
if (input[j+1]<input[j])
{
int temp =input[j+1];
input[j+1]=input[j];
input[j]=temp;
flag = false;
}
}
if (flag) break;
}
if(k>len||k<=0) return arr;
for(int i=0;i<k;i++)
{
arr.push_back(input[i]);
}
return arr;
}
};