AcWing 53 最小的k个数

题目描述：

输入n个整数，找出其中最小的k个数。

注意：

• 数据保证k一定小于等于输入数组的长度;
• 输出数组内元素请按从小到大顺序排序;

样例

输入：[1,2,3,4,5,6,7,8] , k=4

输出：[1,2,3,4]

 分析：

大根堆的经典应用。维护一个大根堆，遍历向量时将元素加入大根堆。一旦大根堆元素个数超过了k，立刻删除堆顶元素，直至遍历到最后一个元素。

class Solution {
public:
    vector getLeastNumbers_Solution(vector input, int k) {
        priority_queue heap;
        for(auto x : input){
            heap.push(x);
            if(heap.size() > k) heap.pop();
        }
        vector v;
        while(heap.size()){
            v.push_back(heap.top());
            heap.pop();
        }
        reverse(v.begin(),v.end());
        return v;
    }
};