使用堆的几种方法
手搓堆
STL::algorithm提供的方法
priority_queue
使用lambda函数自定义复杂结构的比较方法,而不重载<,>
//方法一
auto comp=[](pair<ListNode* ,int> &p1, pair<ListNode* ,int> &p2){
return p1.first->val>p2.first->val;
};
priority_queue<pair<ListNode* ,int>, vector<pair<ListNode* ,int>>, decltype(comp)> heap(comp);
//方法二,加个别名简洁一点
typedef pair<ListNode* ,int> node;
auto comp=[](node &p1, node &p2){
return p1.first->val>p2.first->val;
};
//在c++20以下不能这样使用
typedef pair<ListNode* ,int> node;
priority_queue<node, vector<node>, [](node &p1, node &p2){
return p1.first->val>p2.first->val;
}> heap;
使用仿函数。
仿函数就是将类函数化,即实现类的()运算符,()运算符可以带参数,也可以不带参数。
typedef pair<ListNode* ,int> node;
struct myComp{
bool operator() (node &p1, node &p2){
return p1.first->val>p2.first->val;
}
};
priority_queue<node, vector<node>, myComp> heap;