剑指offer JZ29 最小的K个数（最小堆，以及最大堆）

最新推荐文章于 2025-07-20 12:45:35 发布

peachzy

最新推荐文章于 2025-07-20 12:45:35 发布

阅读量185

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode/剑指offer 文章标签：数据结构剑指offer 最小堆和最大堆 C++ 排序

本文链接：https://blog.youkuaiyun.com/peachzy/article/details/117387788

leetcode/剑指offer 专栏收录该内容

209 篇文章

订阅专栏

这篇博客介绍了如何利用最小堆找到数组中的最小K个数。通过建立一个小顶堆，将前K个元素放入堆中，并在后续元素到来时与堆顶元素比较，实现动态维护最小的K个数。同时，文章还展示了如何创建最大堆，并分别打印了最小堆和最大堆的结果。

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

题目描述：
给定一个数组，找出其中最小的K个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4。如果K>数组的长度，那么返回一个空的数组

思路：
最小堆；最小堆的堆顶是整个堆的最小数

代码如下：

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int>res;
        int n=input.size();
        if(k>n||k==0) return res;
        priority_queue<int,vector<int>>pq;
        for(int val:input){
            if(pq.size()<k){
                pq.push(val);
            }
            else{
                if(val<pq.top()){
                    pq.pop();
                    pq.push(val);
                }
            }
        }
        while(!pq.empty()){
            res.push_back(pq.top());
            pq.pop();
        }
        return res;
    }
};

扩展：最大堆最小堆

#include<iostream>
#include<vector>
#include<queue>
using namespace std;
int main(){
    vector<int>nums;
    nums.push_back(4);
    nums.push_back(5);
    nums.push_back(1);
    nums.push_back(6);
    nums.push_back(2);
    nums.push_back(7);
    nums.push_back(3);
    nums.push_back(8);
    priority_queue<int,deque<int>,greater<int> > q1;//最小堆
    for(int i=0;i<nums.size();i++){
        q1.push(nums[i]);
    }
    while(!q1.empty()){
        cout<<q1.top()<<" ";
        q1.pop();
    }
    cout<<"最小堆"<<endl;
    priority_queue<int> q2;//最大堆
    for(int i=0;i<nums.size();i++){
        q2.push(nums[i]);
    }
    while(!q2.empty()){
        cout<<q2.top()<<" ";
        q2.pop();
    }
    cout<<"最大堆"<<endl;
    cout<<"hello"<<endl; 
}