C++ priority_queue自定义排序_C++优先队列重载比较函数的写法

答案：priority_queue默认为大根堆，通过自定义仿函数可实现小根堆或结构体排序，需重载operator()，返回a.score > b.score使分数低的优先，lambda不能作模板参数，应使用仿函数。

在C++中，priority_queue默认是一个大根堆（最大值优先），但很多时候我们需要自定义排序规则，比如让最小值优先，或者根据结构体的某个成员排序。这就需要重载比较函数。

基本语法与默认行为

priority_queue 的定义如下：

priority_queueainer, Compare>

• T：元素类型
• Container：底层容器，通常用 vector
• Compare：比较函数或仿函数，决定优先级顺序

默认情况下，Compare 是 less，即大顶堆。若使用 greater，则变为小顶堆。

例如，创建一个最小堆：

priority_queue, greater> pq;

自定义比较函数：结构体或类

当元素是结构体时，需自定义比较逻辑。常见做法是定义一个仿函数（重载 operator()）。

示例：按学生的分数升序排列（分数低的优先）

struct Student {
   int id;
   int score;
};

struct Compare {
   bool operator()(const Student& a, const Student& b) {
      return a.score > b.score; // 小顶堆：score 小的优先
   }
};

priority_queue, Compare> pq;

注意：这里返回 a.score > b.score 才能让分数小的排前面。因为 priority_queue 默认取“最大”元素，而我们通过反向比较实现最小值优先。

Lambda 表达式不能直接用于模板参数

以下写法是错误的：

auto cmp = [](const Student& a, const Student& b) {
   return a.score > b.score;
};
priority_queue, decltype(cmp)> pq(cmp); // 错误！

lambda 有唯一类型，不能作为模板参数直接传入。但可以用 function 包装，或改用仿函数。

使用 std::function 和构造函数传参（进阶）

如果想动态传入比较逻辑，可以封装一个类，但标准 priority_queue 不支持运行时传 lambda 作为模板参数。更灵活的方式是使用其他数据结构，或固定使用仿函数。

推荐做法仍是定义结构体仿函数，清晰且高效。

总结关键点

• 默认是大根堆（less），想要小根堆用 greater
• 自定义类型时，写一个仿函数 struct，重载 operator()
• 返回 true 表示 a 应该排在 b 后面（即 a 优先级更低）
• 想要某个字段小的优先，就写 a.field > b.field
• 不要试图把 lambda 当模板参数，会编译失败

基本上就这些。掌握仿函数写法，就能灵活控制优先队列的排序方式了。