问题描述:
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
源码:
解法1:改变了原数组,和上一题差不多,快速排序的思想
class Solution {
public:
int partition(vector<int> &input, int left, int right){
int com = input[left];
while(left<right){
while(left<right && com<=input[right]) right--;
input[left] = input[right];
while(left<right && com>=input[left]) left++;
input[right] = input[left];
}
input[left] = com;
return left;
}
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
vector<int> result;
int length = input.size();
if(input.empty() || k==0 || k>length) return result;
if(k==length) return input;
int left=0, right=length-1, cnt=0, index=0;
while(left<right){
index = partition(input, left, right);
if(index==k-1) break;
else if(index>k-1) right = index-1;
else left = index+1;
}
// copy(input.begin(), input.begin()+k, result.begin());
return vector<int>(input.begin(), input.begin()+k);
}
};
解法2:大顶堆
不改变原数组,用大顶堆的方式,首先简介一下STL中大顶堆的几个函数。
对make_heap(), pop_heap(), push_heap()的用法做个总结:
make_heap()生成堆,他有两个参数,也可以有三个参数,前两个参数是指向开始元素的迭代器和指向结束元素的下一个元素的迭代器。第三个参数是可选的,可以用伪函数less()和greater()来生成大顶堆和小顶堆,其中type为元素类型。如果只传入前两个参数,默认是生成大顶堆。
push_heap()是在堆的基础上进行数据的插入操作,参数与make_heap()相同,需要注意的是,只有make_heap()和push_heap()同为大顶堆或小顶堆,才能插入。
pop_heap()是在堆的基础上,弹出堆顶元素。这里需要注意的是,pop_heap()并没有删除元素,而是将堆顶元素和数组最后一个元素进行了替换,如果要删除这个元素,还需要对数组进行pop_back()操作。
class Solution {
public:
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
int len = input.size();
if(len==0 || k==0 || k>len) return vector<int>();
if(k==len) return input;
vector<int> result(input.begin(), input.begin()+k);
make_heap(result.begin(), result.end());
for(int i=k; i<len; i++){
if(input[i]<result[0]){
pop_heap(result.begin(), result.end());
result.pop_back();
result.push_back(input[i]);
push_heap(result.begin(), result.end());
}
}
return result;
}
};