题解 | #最小的K个数#
最小的K个数
https://www.nowcoder.com/practice/6a296eb82cf844ca8539b57c23e6e9bf
#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; } };