读者可以先看这篇文章,可以为你解答为什么 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);