[剑指offer]最小的K个数

题目描述

输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4,。

分析：寻找最小的k个数，构造一个大小为k最大堆，若k后面的元素比堆顶元素大，直接跳过，若比堆顶元素小，则弹出堆顶元素，把这个元素插入堆。

代码：

class Solution {
public:
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
priority_queue<int> maxheap;
int n=input.size();
if(k<=0||k>n){
return vector<int>();
}
for(int i=0;i<n;i++){
if(maxheap.size()<k){
maxheap.push(input[i]);
}else{
if(input[i]>=maxheap.top()){
continue;
}else{
maxheap.pop();
maxheap.push(input[i]);
}
}
}
vector<int> ret;
while(!maxheap.empty()){
ret.push_back(maxheap.top());
maxheap.pop();
}
return vector<int>(ret.rbegin(),ret.rend());
}
};