默认情况下,C++的std::priority_queue是最大堆,通过使用std::greater<int>可实现基础类型的最小堆;处理自定义类型时,需定义比较结构体,如重载operator()并返回a.age > b.age以实现按年龄升序的最小堆,注意lambda不能直接用于模板参数。

默认情况下,C++ 的 std::priority_queue 是一个最大堆(即顶部元素是最大的)。如果需要实现最小堆,可以通过自定义比较器来改变其行为。
使用 std::greater 实现最小堆
最简单的方法是使用 std::greater 作为比较函数对象。适用于基础类型如 int、double 等。
#include <vector>
#include <iostream>
#include <functional> // std::greater
示例代码:
std::priority_queue<int, std::vector<int>, std::greater<int>> min_heap;min_heap.push(10);
min_heap.push(5);
min_heap.push(20);
while (!min_heap.empty()) {
std::cout << min_heap.top() << " ";
min_heap.pop();
}
// 输出: 5 10 20
自定义结构体 + 自定义比较器
当处理自定义类型时,比如结构体或类,可以定义自己的比较逻辑。
立即学习“C++免费学习笔记(深入)”;
struct Person {std::string name;
int age;
Person(std::string n, int a) : name(n), age(a) {}
};
希望按年龄从小到大排序(最小堆),可以写仿函数(函数对象):
struct CompareAge {bool operator()(const Person& a, const Person& b) {
return a.age > b.age; // 注意:这里用 > 实现最小堆
}
};
std::priority_queue<Person, std::vector<Person>, CompareAge> pq;
示例使用:
pq.push(Person("Alice", 30));pq.push(Person("Bob", 25));
pq.push(Person("Charlie", 35));
while (!pq.empty()) {
std::cout << pq.top().name << " " << pq.top().age << "\n";
pq.pop();
}
// 输出: // Bob 25 // Alice 30 // Charlie 35
Lambda 表达式不能直接用于模板参数
注意:你不能直接把 lambda 写进 priority_queue 的模板参数里,因为模板需要类型,而每个 lambda 都是唯一类型且无法命名。但可以用 std::function 包装或改用函数指针(不推荐复杂场景)。
更灵活的方式仍是使用仿函数或 struct 重载 operator()。
基本上就这些。用 std::greater 快速实现基础类型的最小堆,复杂类型自己写比较结构体即可。关键点是:想让小的优先级高,就在比较中返回 a > b,这样系统认为 a 应该排在后面,从而维持堆顶最小。







