首页 > 后端开发 > C++ > 正文

C++怎么实现一个堆(Heap)_C++数据结构与优先队列(priority_queue)

php中文网
发布: 2025-12-07 19:16:02
原创
658人浏览过
C++中堆是基于完全二叉树的结构,用于实现优先队列。1. STL的priority_queue默认为最大堆,提供push、top、pop等操作;2. 手动实现需掌握shiftUp和shiftDown以维护堆序性;3. 堆适用于优先队列、Top K问题、堆排序和Dijkstra算法;4. 注意priority_queue不支持遍历,手动实现时防止数组越界,自定义类型需重载比较规则。

c++怎么实现一个堆(heap)_c++数据结构与优先队列(priority_queue)

在C++中,堆(Heap)是一种基于完全二叉树的数据结构,常用于实现优先队列。堆分为最大堆(大根堆)和最小堆(小根堆),其中最大堆的父节点值不小于子节点,最小堆则相反。C++标准库提供了 priority_queue 来方便使用堆,但理解手动实现堆有助于掌握其底层原理。

1. 使用 STL 的 priority_queue 实现堆

C++ 标准库中的 priority_queue 默认实现的是最大堆,基于 vector 和堆算法自动维护堆序性。

基本用法:

  • priority_queue<int> max_heap;</int>:创建最大堆
  • priority_queue<int vector>, greater<int>> min_heap;</int></int>:创建最小堆
  • push(x):插入元素
  • top():获取堆顶元素
  • pop():删除堆顶元素
  • empty()size():判断是否为空和获取大小

示例代码:

立即学习C++免费学习笔记(深入)”;

#include <iostream>
#include <queue>
using namespace std;
<p>int main() {
priority_queue<int> max_heap;
max_heap.push(10);
max_heap.push(30);
max_heap.push(20);</p><pre class="brush:php;toolbar:false;">while (!max_heap.empty()) {
    cout << max_heap.top() << " ";
    max_heap.pop();
}
// 输出:30 20 10
return 0;
登录后复制

}

2. 手动实现最大堆

手动实现堆可以加深对上浮(shift up)和下沉(shift down)操作的理解。通常使用数组存储完全二叉树。

关键操作:

Beautiful.ai
Beautiful.ai

AI在线创建幻灯片

Beautiful.ai 108
查看详情 Beautiful.ai
  • 插入(push):将元素添加到末尾,然后执行上浮操作维护堆性质
  • 删除堆顶(pop):将最后一个元素移到堆顶,然后执行下沉操作
  • 上浮(shiftUp):比较当前节点与父节点,若大于父节点则交换
  • 下沉(shiftDown):比较父节点与两个子节点,与较大者交换直到满足堆性质

简单实现示例:

#include <iostream>
#include <vector>
using namespace std;
<p>class MaxHeap {
private:
vector<int> heap;</p><pre class="brush:php;toolbar:false;">void shiftUp(int index) {
    while (index > 0) {
        int parent = (index - 1) / 2;
        if (heap[index] <= heap[parent]) break;
        swap(heap[index], heap[parent]);
        index = parent;
    }
}

void shiftDown(int index) {
    int n = heap.size();
    while (index < n) {
        int left = 2 * index + 1;
        int right = 2 * index + 2;
        int maxIndex = index;

        if (left < n && heap[left] > heap[maxIndex])
            maxIndex = left;
        if (right < n && heap[right] > heap[maxIndex])
            maxIndex = right;

        if (maxIndex == index) break;
        swap(heap[index], heap[maxIndex]);
        index = maxIndex;
    }
}
登录后复制

public: void push(int val) { heap.push_back(val); shiftUp(heap.size() - 1); }

void pop() {
    if (heap.empty()) return;
    heap[0] = heap.back();
    heap.pop_back();
    if (!heap.empty())
        shiftDown(0);
}

int top() {
    return heap.empty() ? -1 : heap[0];
}

bool empty() {
    return heap.empty();
}

int size() {
    return heap.size();
}
登录后复制

};

这个类实现了基本的最大堆功能,可用于替代 priority_queue 理解内部机制。

3. 堆的应用场景

堆常用于以下场景:

  • 优先队列:任务调度、事件处理等需要按优先级出队的场合
  • 求 Top K 元素:例如找出最大或最小的 K 个数,使用大小为 K 的堆效率高
  • 堆排序:时间复杂度 O(n log n),原地排序
  • Dijkstra 算法:结合最小堆可高效提取最短路径节点

例如,找数组中最大的 K 个数,可以用最小堆维护 K 个元素,遍历过程中只保留较大的值。

4. 注意事项

使用堆时需要注意:

  • STL 的 priority_queue 不支持遍历和删除非堆顶元素
  • 手动实现时注意数组越界,特别是左右子节点索引计算
  • 自定义类型需重载比较函数或提供仿函数
  • 堆的插入和删除时间复杂度为 O(log n),建堆过程可优化至 O(n)

基本上就这些。掌握 priority_queue 的使用和堆的手动实现,能更好应对算法题和实际开发中的优先级管理需求。堆的核心在于维护堆序性,理解 shiftUp 和 shiftDown 是关键。

以上就是C++怎么实现一个堆(Heap)_C++数据结构与优先队列(priority_queue)的详细内容,更多请关注php中文网其它相关文章!

最佳 Windows 性能的顶级免费优化软件
最佳 Windows 性能的顶级免费优化软件

每个人都需要一台速度更快、更稳定的 PC。随着时间的推移,垃圾文件、旧注册表数据和不必要的后台进程会占用资源并降低性能。幸运的是,许多工具可以让 Windows 保持平稳运行。

下载
来源:php中文网
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
最新问题
开源免费商场系统广告
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板
关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新 English
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送
PHP中文网APP
随时随地碎片化学习

Copyright 2014-2025 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号