在之前的blog中介绍过STL库中set、map、priority_queue三种排序方式,那么相信读者会经常用到特殊的type,这时,我们需要重写compartor,举个栗子
以Pair中第二个元素构建大根堆
auto cmp = [](const pair<int, int> &a, const pair<int, int> &b) {
return a.second < b.second;//less
};
priority_queue<pair<int, int>, vector<pair<int, int>>, decltype(cmp)>q(cmp);
注意:
q(cmp)
;因为lambda这种特殊的class没有默认构造函数,priority_queue内部排序比较的时候要使用的是一个实例化的lambda对象,只能通过lambda的 copy构造进行实例化,从哪里copy呢,就需要priority_queue构造函数的时候传入这个lambda对象。priority_queue的自定义比较也可以使用struct或者class,因为struct和class都有默认构造函数,此时就不需要pq(Type) 而是直接pq即可。同理,自定义比较也可以使用函数,同样此时也需要提供函数对象进行实例化pq(CmpFunc),假设比较函数为bool CmpFunc(int x, int y) {return x < y;}
同理,若想要构建小根堆,则只需将lambda表达式中的比较条件更改即可
以Pair中第二个元素构建小根堆
auto cmp = [](const pair<int, int> &a, const pair<int, int> &b) {
return a.second > b.second;//greater
};
priority_queue<pair<int, int>, vector<pair<int, int>>, decltype(cmp)>q(cmp);
除此以外,我们还可以使用仿函数(函数对象),在这里仅给出小根堆
class compare {
public:
bool operator()(const pair<int, int> &a, const pair<int, int> &b) {
return a.second > b.second;//greater
}
};
priority_queue<pair<int, int>, vector<pair<int, int>>, compare>q;