STL priority_queue自定义排序实现

首先区分一下sort和priority_queue的cmp函数

sort()函数的cmp为函数，priority_queue的cmp为函数对象

sort()中的cmp写法（升序）

static bool cmp(vector<int>& a, vector<int>& b){
        return a[1] < b[1];
    }

priority_queue中的cmp写法（升序）

struct cmp{
    bool operator() (ListNode* p1, const ListNode* p2){
        return p1->val > p2->val;
    }
};

————————————————————————————————————————

当使用优先队列适配器是，如果我们想根据自己的需求实现排序，该怎么做呢？

具体问题可见leetcode 23 合并K个升序链表

函数对象

首先来回顾一下什么叫做函数对象？

函数对象就是一个重载了（）运算符的类的对象，这样就可以直接使用‘对象()’的方式，就像调用函数一样。

在STL中，priority_queue的定义如下，其中第三个参数std::less<T> 和 std::greater<T> 都是以函数对象的方式定义在 <function> 头文件中。

template <typename T,
        typename Container=std::vector<T>,
        typename Compare=std::less<T> >

方法一 使用自定义函数对象

无论基础数据类型，还是类对象数据类型，都可以使用函数对象自定义的排序规则

template <typename T>
class cmp
{
public:
    //重载 () 运算符
    bool operator()(T a, T b)
    {
        return a > b;
    }
};

priority_queue<int,vector<int>,cmp<int> > pq(a,a+5);

如上文所示，通过cmp类（或者结构体）重载的（）运算符中定义排序规则，并将其实例化后作为priority_queue的第3个参数传入，即可实现自定义比较函数。

方法二 重载<或>

class node {
public:
    node(int x = 0, int y = 0) :x(x), y(y) {}
    int x, y;
};
//新的排序规则为：先按照 x 值排序，如果 x 相等，则按 y 的值排序
bool operator < (const node &a, const node &b) {
    if (a.x > b.x) return 1;
    else if (a.x == b.x)
        if (a.y >= b.y) return 1;
    return 0;
}

这样实际上还是使用的 std::less<T>