std::priority_queue是C++标准库中的优先队列容器,默认情况下它会按照元素从大到小的顺序出队,也就是大顶堆的实现。但实际开发中我们往往需要自定义优先级规则,比如实现小顶堆、按照自定义结构体的某个成员排序等,下面汇总几种常用的自定义优先级方法。

方法一:重载结构体的比较运算符
如果优先队列存储的是自定义结构体,我们可以直接在结构体内部重载<运算符,来定义优先级规则。默认情况下优先队列使用<来判断优先级,所以重载后的规则会直接生效。
下面的示例定义了一个任务结构体,我们希望按照任务的优先级数值从小到大出队,也就是小顶堆的效果:
#include <iostream>
#include <queue>
#include <vector>
// 自定义任务结构体
struct Task {
int id;
int priority; // 优先级数值,越小优先级越高
// 重载<运算符,定义优先级规则
bool operator<(const Task& other) const {
// 注意:priority_queue默认是大顶堆,所以这里要反过来写
// 当当前任务的优先级大于其他任务时,当前任务优先级更低
return priority > other.priority;
}
};
int main() {
std::priority_queue<Task> pq;
pq.push({1, 3});
pq.push({2, 1});
pq.push({3, 2});
while (!pq.empty()) {
Task t = pq.top();
pq.pop();
std::cout << "任务ID: " << t.id << ", 优先级: " << t.priority << std::endl;
}
return 0;
}
运行上述代码,输出结果会按照优先级数值1、2、3的顺序出队,符合我们的预期。
方法二:使用仿函数作为比较器
如果不想修改结构体本身,或者需要多种不同的排序规则,可以自定义仿函数作为优先队列的比较器。仿函数本质是重载了()运算符的结构体或类。
下面的示例同样实现小顶堆的效果,但是不修改Task结构体:
#include <iostream>
#include <queue>
#include <vector>
struct Task {
int id;
int priority;
};
// 自定义仿函数,作为比较器
struct CompareTask {
bool operator()(const Task& a, const Task& b) const {
// 返回true表示a的优先级比b低
return a.priority > b.priority;
}
};
int main() {
// 声明优先队列时传入比较器类型
std::priority_queue<Task, std::vector<Task>, CompareTask> pq;
pq.push({1, 3});
pq.push({2, 1});
pq.push({3, 2});
while (!pq.empty()) {
Task t = pq.top();
pq.pop();
std::cout << "任务ID: " << t.id << ", 优先级: " << t.priority << std::endl;
}
return 0;
}
方法三:使用lambda表达式作为比较器
C++11之后支持lambda表达式,我们可以用lambda来定义比较规则,这种方式更加灵活,适合临时定义简单的排序逻辑。
需要注意的是,lambda表达式作为比较器时,需要先在外部定义lambda变量,再传入优先队列的模板参数中:
#include <iostream>
#include <queue>
#include <vector>
#include <functional>
struct Task {
int id;
int priority;
};
int main() {
// 定义lambda比较函数
auto cmp = [](const Task& a, const Task& b) {
return a.priority > b.priority;
};
// 声明优先队列,使用decltype获取lambda的类型
std::priority_queue<Task, std::vector<Task>, decltype(cmp)> pq(cmp);
pq.push({1, 3});
pq.push({2, 1});
pq.push({3, 2});
while (!pq.empty()) {
Task t = pq.top();
pq.pop();
std::cout << "任务ID: " << t.id << ", 优先级: " << t.priority << std::endl;
}
return 0;
}
方法四:内置类型的小顶堆实现
如果优先队列存储的是int、double等内置类型,想要实现小顶堆,不需要自定义结构体,可以直接使用标准库的std::greater比较器:
#include <iostream>
#include <queue>
#include <vector>
#include <functional>
int main() {
// 小顶堆,使用std::greater作为比较器
std::priority_queue<int, std::vector<int>, std::greater<int>> pq;
pq.push(3);
pq.push(1);
pq.push(2);
while (!pq.empty()) {
std::cout << pq.top() << " ";
pq.pop();
}
// 输出:1 2 3
return 0;
}
不同方法的适用场景
我们可以根据需求选择合适的方法:
- 如果自定义结构体只有一种固定的排序规则,优先选择重载比较运算符的方式,代码更简洁。
- 如果需要多种不同的排序规则,或者不想修改结构体本身,选择仿函数的方式,可复用性更强。
- 如果是临时使用的简单排序逻辑,选择lambda表达式的方式,代码更紧凑。
- 如果是内置类型的小顶堆需求,直接使用
std::greater即可,不需要额外定义比较规则。
注意事项
在使用自定义优先级时,需要注意优先队列的比较规则是:比较函数返回true时,第一个参数的优先级低于第二个参数。很多开发者会搞反这个逻辑,导致排序结果不符合预期。另外,自定义比较器时需要保证比较规则是严格弱序的,否则可能会出现未定义行为。
std::priority_queue自定义优先级比较函数仿函数lambda表达式修改时间:2026-07-13 21:12:26