求最小k个数

最新推荐文章于 2024-08-27 14:41:49 发布

瘪瘪前端加油

最新推荐文章于 2024-08-27 14:41:49 发布

阅读量88

点赞数

分类专栏：算法专栏

本文链接：https://blog.csdn.net/weixin_41682025/article/details/111639902

版权

算法专栏专栏收录该内容

1 篇文章 0 订阅

订阅专栏

class Solution {
public:
    vector<int> smallestK(vector<int>& arr, int k) {
        vector<int> res;
        priority_queue<int> q;//默认为大顶堆，用greater就是小顶堆
        for(int a : arr) {
            q.push(a);
            if(q.size() > k) {
                q.pop();
            }
        }
        while(!q.empty()) {
            res.push_back(q.top());
            q.pop();
        }
        return res;
    }
};