LeetCode刷题笔记——40. 最小的k个数（最大堆、priority_queue、vector）

本文链接：https://blog.csdn.net/cxxxxxxxxxxxxx/article/details/106843730

这篇博客介绍了如何使用最大堆、C++的priority_queue和vector来解决LeetCode上的‘最小的k个数’问题。通过先排序再取前k个数，或者利用大根堆动态维护最小的k个数，提供了详细的解题思路。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目

在这里插入图片描述

1. 最大堆

2. priority_queue

3. vector

4. 答题

参考来源：https://leetcode-cn.com/problems/zui-xiao-de-kge-shu-lcof/solution/zui-xiao-de-kge-shu-by-leetcode-solution/

对原数组从小到大排序后取出前 k个数即可。

class Solution {
public:
    vector<int> getLeastNumbers(vector<int>& arr, int k) {
        vector<int> vec(k, 0);
        sort(arr.begin(), arr.end());//sort from small to big
        for(int i = 0; i < k; i++)
            vec[i] = arr[i];
        return vec;
    }
};

在这里插入图片描述
2. 我们用一个大根堆实时维护数组的 k个小值。首先将前 k 个数插入大根堆中，随后从第 k+1个数开始遍历，如果当前遍历到的数比大根堆的堆顶的数要小，就把堆顶的数弹出，再插入当前遍历到的数。最后将大根堆里的数存入数组返回即可。

class Solution {
public:
    vector<int> getLeastNumbers(vector<int>& arr, int k) {
        vector<int> vec(k, 0);
        if(k==0)   //排除k为0的情况
            return vec;
        priority_queue<int> Q; //实例化最大堆
        for(int i = 0; i < k; i++)
            Q.push(arr[i]); //先push进arr前k个数
        for(int i = k; i < (int)arr.size(); i++)//处理k后面的数字
        {
            if(Q.top() > arr[i])
            {
                Q.pop();
                Q.push(arr[i]);
            }
        }
        for(int i = 0; i < k; i++){
            vec[i] = Q.top();
            Q.pop();//输出top并pop
        }
        return vec；
    }
};

在这里插入图片描述