题解 | #最小的K个数#

#include <functional>
#include <queue>
#include <vector>
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param input int整型vector 
     * @param k int整型 
     * @return int整型vector
     */
     //priority_queue<int,vector<int>,greater<>>优先级队列，按照一定优先级排序的队列，第一个参数为数据类型，
     //第二个为存储容器结构，第三个为优先级，less为降序排列，即大顶堆。
    vector<int> GetLeastNumbers_Solution(vector<int>& input, int k) {
        // write code here
        if (k>input.size()) {
            return vector<int>();
        }
        priority_queue<int,vector<int>,greater<>> pq;
        for (auto a: input) {
            pq.push(a);
        }
        vector<int> re;
        while (k--) {
            re.push_back(pq.top());
            pq.pop();
        }
        return re;
    }
};