C/C++｜关于 std::priority_queue （优先级队列）的第三个模版参数

Che3rry2024-08-14 8:48

读者可以先看这篇文章，可以为你解答为什么 std::priority_queue 的第三个模版参数这么讲究。

总而言是，第三个参数要求的是一个比较器对象

被称之为对象，那么肯定是一个类，并且该类应该可以被实例化为一个对象。最好理解的就是仿函数（重载 operator() 的类），这里总结几套写法

自定义函数对象：

cpp 复制代码

struct Compare {
    bool operator()(ListNode* a, ListNode* b) {
        return a->val > b->val;
    }
};
std::priority_queue<ListNode*, std::vector<ListNode*>, Compare> pq;

Lambda表达式

cpp 复制代码

auto cmp = [](ListNode* a, ListNode* b) { return a->val > b->val; };
std::priority_queue<ListNode*, std::vector<ListNode*>, decltype(cmp)> pq(cmp);

当然了，我们也可以使用 std::function 直接声明 cmp 的类型：

cpp 复制代码

std::priority_queue<ListNode*, 
					std::vector<ListNode*>, 
					std::function<bool>(ListNode*, ListNode*)> pq(cmp);

函数指针

cpp 复制代码

bool compare(ListNode* a, ListNode* b) {
    return a->val > b->val;
}
std::priority_queue<ListNode*, 
					std::vector<ListNode*>, 
					bool(*)(ListNode*, ListNode*)> pq(compare);