C++中priority_queue默认实现最大堆,通过指定greater比较器可实现最小堆,支持自定义类型及比较逻辑,简化堆操作。

在C++中,堆(Heap)是一种特殊的完全二叉树结构,分为最大堆和最小堆。最大堆的根节点值最大,最小堆的根节点值最小。虽然可以手动实现堆的插入、删除等操作,但C++标准库提供了更简便的方式——priority_queue,它默认实现的是最大堆。
使用 priority_queue 实现最大堆
priority_queue 默认基于 vector 和 less 比较器,因此顶部元素是最大的,即最大堆。
- #include <queue>
- #include <iostream>
- std::priority_queue<int> max_heap;
- max_heap.push(10);
- max_heap.push(30);
- max_heap.push(20);
- std::cout << max_heap.top(); // 输出 30
每次调用 top() 获取当前最大值,pop() 删除最大值,所有操作自动维护堆结构。
使用 priority_queue 实现最小堆
要实现最小堆,需指定第三个模板参数为 greater,并明确容器类型。
立即学习“C++免费学习笔记(深入)”;
- #include <queue>
- #include <vector>
- #include <functional> // std::greater
- std::priority_queue<int, std::vector<int>, std::greater<int>> min_heap;
- min_heap.push(30);
- min_heap.push(10);
- min_heap.push(20);
- std::cout << min_heap.top(); // 输出 10
此时堆顶始终是最小元素,适用于需要频繁获取最小值的场景,比如Dijkstra算法。
自定义类型如何使用 priority_queue
若堆中存储的是自定义结构体或类,需提供比较逻辑。可通过重载操作符或传入仿函数。
- struct Person {
- int age;
- std::string name;
- };
- // 最小堆:按年龄排序
- auto cmp = [](const Person& a, const Person& b) { return a.age > b.age; };
- std::priority_queue<Person, std::vector<Person>, decltype(cmp)> pq(cmp);
使用lambda表达式作为比较器时,需将其实例传入构造函数,并在模板中声明类型。
基本上就这些。priority_queue 封装了堆的核心操作,避免手动调整堆结构,提升开发效率。理解其默认行为和如何切换最小堆,能灵活应对各类算法需求。不复杂但容易忽略细节,比如最小堆必须显式指定容器和比较器。









