C++关于sort和priority_queue的运算符重载

C++中的sort函数默认是将元素升序排列的，而priority_queue默认是将元素降序排列的（默认实现的是大顶堆）。

自定义运算符用的比较多，以下2种对sort和priority_queue运算符的重载都有效，效果都是一样的：

 #include <iostream>
 #include <vector>
 #include <algorithm>
 #include <queue>
 using namespace std;

 //sort实现的都是先按a值降序排列，a相同时，按b值降序排列
 //priority_queue和sort的默认排序相反，实现的都是先按a值升序排列，a相同时，按b值升序排列

 //重载方式一：
 struct Node {
     int a, b;
     Node(int x, int y) {
         a = x;
         b = y;
     }
     bool operator < (const Node &c) const {
         if(a == c.a) return b > c.b;
         return a > c.a;
     }
 };

 //重载方式二：注意要有public
 // class Node {
 //     public:
 //     int a, b;
 //     Node(int x, int y) {
 //         a = x;
 //         b = y;
 //     }
 //     bool operator < (const Node &c) const {
 //         if(a == c.a) return b > c.b;
 //         return a > c.a;
 //     }
 // };
 int main() {
     vector<Node> v;
     v.push_back(Node(,));
     v.push_back(Node(,));
     v.push_back(Node(,));
     v.push_back(Node(,));
     sort(v.begin(), v.end());
     for(int i = ; i < v.size(); ++i) {
         cout<<v[i].a<<" "<<v[i].b<<endl;
     }
     cout<<endl;

     priority_queue<Node> pq;
     pq.push(Node(,));
     pq.push(Node(,));
     pq.push(Node(,));
     pq.push(Node(,));
     while(!pq.empty()) {
         cout<<pq.top().a<<" "<<pq.top().b<<endl;
         pq.pop();
     }
     return ;
 }

输出

对sort单独自定义运算符：

 #include <iostream>
 #include <vector>
 #include <algorithm>
 #include <queue>
 using namespace std;

 //sort实现的都是先按a值降序排列，a相同时，按b值降序排列
 //priority_queue和sort的默认排序相反，实现的都是先按a值升序排列，a相同时，按b值升序排列

 struct Node {
     int a, b;
     Node(int x, int y) {
         a = x;
         b = y;
     }
 };
 /*
 //重载方式一：
 struct compare {
     bool operator()(const Node a, const Node b) {
         if(a.a == b.a) return a.b > b.b;
         return a.a > b.a;
     }
 } cmp;
 */

 /*
 //重载方式二：
 bool cmp(const Node a, const Node b) {
     if(a.a == b.a) return a.b > b.b;
     return a.a > b.a;
 }
 */

 int main() {

     vector<Node> v;
     v.push_back(Node(,));
     v.push_back(Node(,));
     v.push_back(Node(,));
     v.push_back(Node(,));
     //重载方式一和二：
     sort(v.begin(), v.end(), cmp);
     //重载方式三：
     sort(v.begin(), v.end(), [](const Node a, const Node b) {
         if(a.a == b.a) return a.b > b.b;
         return a.a > b.a;
     });
     for(int i = ; i < v.size(); ++i) {
         cout<<v[i].a<<" "<<v[i].b<<endl;
     }
     cout<<endl;
     return ;
 }

输出都是

对priority_queue单独自定义运算符：

#include <iostream>
#include <vector>
#include <algorithm>
#include <queue>
using namespace std;

//sort实现的都是先按a值降序排列，a相同时，按b值降序排列
//priority_queue和sort的默认排序相反，实现的都是先按a值升序排列，a相同时，按b值升序排列

struct Node {
    int a, b;
    Node(int x, int y) {
        a = x;
        b = y;
    }
};

struct cmp {
    bool operator()(const Node a, const Node b) {
        if(a.a == b.a) return a.b > b.b;
        return a.a > b.a;
    }
};

int main() {
    priority_queue<Node, vector<Node>, cmp> pq;

    pq.push(Node(,));
    pq.push(Node(,));
    pq.push(Node(,));
    pq.push(Node(,));
    while(!pq.empty()) {
        cout<<pq.top().a<<" "<<pq.top().b<<endl;
        pq.pop();
    }

    return ;
}

效果：