题目描述
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4。
解法:1.进行数组的分区,返回分区index值
2.根据index和k-1的大小来判断是否已经获取了足够的数,否则递归调用即可
3.分区函数的编写,快速排序的思想
class Solution {
public:
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
int len = input.size();
vector<int> output;
if(len == 0 || k == 0 || k > len) return output;
int start = 0;
int end = len - 1;
int index = partion(input, start, end);
while(index != k - 1) { //k个数,但是最后一个下标为k-1
if(index > k - 1) { //说明范围还是太大了,进行缩小
end = index - 1;
index = partion(input, start, end);
}
else {
start = index + 1;
index = partion(input, start, end);
}
}
for(int i = 0; i < k; ++i)
output.push_back(input[i]);
return output;
}
int partion(vector<int> &input, int start, int end) {
int len = input.size();
if(len == 0) return 0;
int small = start - 1;
int index = (start + end) / 2;
swap(input[end], input[index]); //把比较值放到最后去
for(int index = start; index < end; ++index) {
if(input[index] < input[end]) {//如果指向的值小于比较值,则继续移动
++small;
if(index != small) //如果不等于的话,就是当前small指向的值较大,与后面的index互换
swap(input[index], input[small]);
}
}
++small; //此时small才是指向较大的值,与最后的一个值进行互换
swap(input[end], input[small]);
return small;
}
};