C++ 自定义排序与优先队列运算符重载

方案 3：小根堆 (工程正统流 —— 配合 greater)

• 场景：工程代码、团队协作、需要显式语义。
• 原理：替换裁判。将比较器从默认的 less 替换为 greater。
• 要求：结构体必须重载 operator>。

struct node{
    int z;
    // 逻辑：诚实重载大于号，不欺骗
    friend bool operator>(node a,node b){ return a.z>b.z; }
};
// 声明：显式指定 greater，并填补 vector 占位符
priority_queue<node,vector<node>,greater<node>> q;

三、核心原理解析：参数顺序与逻辑陷阱 (必读)

在重载 friend bool operator<(Type a, Type b) 时，你给参数起名 x 还是 y 无所谓，但参数的位置决定了谁是比较符号 < 左边的数，谁是右边的数。这是一个极易混淆的基础概念。

1. 核心规则：位置决定身份

当你写 friend bool operator < (Node a, Node b) 时，C++ 编译器遵循以下铁律：

• 第 1 个参数 (a)：代表 小于号左边 的那个对象 (LHS, Left Hand Side)。
• 第 2 个参数 (b)：代表 小于号右边 的那个对象 (RHS, Right Hand Side)。

当我们执行逻辑 x < y 时，编译器实际调用的是 operator<(x, y)。

2. 举例演示：张三 vs 李四

假设我们有两个病人，按等级 (level) 排序：

• 张三 (Left)：level = 10
• 李四 (Right)：level = 20
• 比较操作：张三 < 李四

情况 A：正常写法 (符合直觉)

friend bool operator < (people a, people b){ // a 是张三 (10)，b 是李四 (20)
    return a.level < b.level; // 逻辑：10 < 20 -> 返回 true
}

• 比较结果：true (张三比李四'小')。
• 优先队列行为：priority_queue 判定 B (李四) 比 A (张三) 强。
• 最终效果：大根堆。强者 (李四 20) 上浮至堆顶。

情况 B：逻辑反转 (参数逻辑写反)

如果你搞混了顺序，或者故意写反：

friend bool operator < (people a, people b){ // a 依然是张三 (10)，b 依然是李四 (20)
    return a.level > b.level; // <--- 逻辑反转！
    // 逻辑：10 > 20 -> 返回 false
}

• 比较结果：false (张三不比李四'小'，在队列眼里张三比李四'强')。
• 最终效果：小根堆。被视为'强'的张三 (10) 上浮至堆顶。

3. 为什么优先队列会受此影响？

std::priority_queue 的底层逻辑非常简单粗暴：

1. 它不断调用 operator<(A, B)。
2. 如果返回 true：它认为 B 的优先级比 A 高 (B is strict weak ordering greater than A)。
3. 动作：它把 B 往堆顶调整。

结论推导：

• return a.val < b.val; -> 数值大的优先级高 -> 大数在顶。
• return a.val > b.val; -> 数值小的优先级高 -> 小数在顶。

4. 万能公式

为了符合直觉，永远建议按标准写法来，需要小根堆时再去配合 greater 或修改逻辑，不要乱换参数位置：

// 标准模版
friend bool operator < (const Node& a, const Node& b) {
    return a.属性 < b.属性; // 正常的升序逻辑 -> 对应大根堆
}

四、greater 与 std::sort 的深度用法

std::greater 是一个模板结构体（仿函数），其本质是调用类型的 operator>。

1. std::sort 的默认行为

默认使用 <，执行升序排列。

vector<int> v={1,5,3};
sort(v.begin(),v.end()); // 结果：1, 3, 5

2. 使用 greater 改为降序

传入 greater 实例，执行降序排列。

// 语法：greater<int>() 是构造一个临时对象
sort(v.begin(),v.end(),greater<int>()); // 结果：5, 3, 1

3. 结构体数组排序 (配合重载 >)

若对自定义结构体使用 greater 排序，结构体内部必须重载 >。

struct node{
    int z;
    // sort 降序需要 > 运算符支持
    friend bool operator>(node a,node b){ return a.z>b.z; }
};
vector<node> arr={{1},{5},{3}};
// 自动调用 operator>，实现 z 从大到小
sort(arr.begin(),arr.end(),greater<node>()); // 结果：5, 3, 1

对比记忆表：

五、多关键字排序与自定义仿函数 (进阶)

处理复杂逻辑（如：分数优先，ID 兜底）的两种核心范式。

场景设定

struct Student{
    int score,id;
};

目标：分数 (score) 高的优先；分数相同，学号 (id) 小的优先。

写法 A：结构体内部重载 (推荐比赛)

在一个 operator< 中处理所有层级逻辑。

struct Student{
    int score,id;
    friend bool operator<(Student a,Student b){
        // 第一关键字：分数。分数高的在顶 (大根堆逻辑)
        // 若分数不相等，直接按分数定胜负
        if(a.score!=b.score) return a.score<b.score;
        // 第二关键字：学号 (仅当分数相等时执行)
        // 想要 id 小的在顶，需反转逻辑 (小根堆逻辑)
        return a.id>b.id;
    }
};
priority_queue<Student> q;

写法 B：外部仿函数 (工程推荐/解耦)

不修改结构体源码，通过定义外部'裁判类'控制排序。此方法可实现一套数据结构多种排序方式。

// 纯净数据结构
struct Student{
    int score,id;
};

// 裁判 A：按分数优先
struct CmpScore{
    bool operator()(Student a,Student b){
        if(a.score!=b.score) return a.score<b.score;
        return a.id>b.id;
    }
};

// 裁判 B：纯按 ID 小的优先
struct CmpID{
    bool operator()(Student a,Student b){
        return a.id>b.id; // 小根堆逻辑
    }
};

// 声明时传入具体裁判类型
priority_queue<Student,vector<Student>,CmpScore> q1;
priority_queue<Student,vector<Student>,CmpID> q2;

六、速查表 (多条件逻辑映射)

假设 Key1 为主键，Key2 为次键。基于默认 priority_queue (大根堆)。

心法：

• 想让 大 的在顶：用 < (顺应默认)。
• 想让 小 的在顶：用 > (逻辑反转)。

七、避坑指南

1. 模板与对象的区别 (括号问题)
   • priority_queue 需要的是类型 (Type)：priority_queue<..., Cmp> (无括号)。
   • sort 需要的是对象实例 (Instance)：sort(..., Cmp()) (有括号)。
2. greater 的依赖性
   • 使用 greater<T> 时，T 必须重载 operator>。
   • 报错 no match for operator> 通常是因为只写了 < 却用了 greater。
3. 引用与 Const
   • Friend （友元函数）写法：friend bool operator<(node a,node b) (值传递，简单)。
   • Member （成员函数）写法：bool operator<(const node& a) const (必须加 const，否则 STL 报错)。
   • 建议：始终坚持使用 Friend 写法，语法负担最小。