题解 | #最小的K个数#

#include <pthread.h>
#include <queue>
class Solution {
public:
    /**
     * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
     *
     * 
     * @param input int整型vector 
     * @param k int整型 
     * @return int整型vector
     */
    bool operator()(const int & a, const int & b){ // 小顶堆比较大小
        return a >b ;
    }
    vector<int> GetLeastNumbers_Solution(vector<int>& input, int k) {
        // write code here
        priority_queue<int> priQue; 
        for(int val : input){
            priQue.push(val);
            if(priQue.size() > k){ //若小顶堆大小大于k，则弹出
                priQue.pop();
            }

        }
        vector<int> result;
        while(k--){// 保存结果
            result.push_back(priQue.top());
            priQue.pop();
        }
        return result;
    }
};