C++优先级队列priority_queue自定义比较函数

1.自定义数据类型时

https://blog.csdn.net/HermitSun/article/details/107101944

 参照378. 有序矩阵中第 K 小的元素

复制代码

class Point
{
    int val, x, y;
    Point(int val, int x, int y) : val(val), x(x), y(y) {}
    bool operator>(const Point &p) const { return val > p.val; }
};

复制代码

自定义 定义一个小顶堆,这里的greater运算符函数,参数需要是const,并且需要时const函数,不然会报错:

this‘ argument has type ‘const xxx‘, but method is not marked const

 初始化时:

priority_queue<Point, vector<Point>, greater<Point>> q;

  q.emplace(1, 1, 1);

greater函数的文档:

也就是说在堆排序时,调用greater,greater的参数是const的,并且是const成员函数,在内部比较时会用到上面重写的operator>函数,而const成员函数不能调用非const成员函数,所以会报错,所以需要将其声明为const函数,另外如果operator>的参数列表不声明为const,但operator中为const会导致形参无法赋值。

2.三种方法

https://www.cnblogs.com/Harley-Quinn/p/6104306.html

2.1函数指针

bool greaterInt(const int& lhs, const int& rhs)
{
    return lhs > rhs;
}
priority_queue<int, vector<int>, bool(*)(const int&, const int&)> q(greaterInt);

但这样写其实是写的复杂了,C风格的,不常用。

 2.2函数对象

仿函数类

复制代码

template <typename T>
struct Greater
{
    bool operator()(const T& lhs, const T& rhs) const
    {
        return lhs > rhs;
    }
};
priority_queue<int, vector<int>, Greater<int>> q;

复制代码

采用显式写法:

priority_queue<int, vector<int>, Greater<int>> q(Greater<int>());

即显式地调用Greater类的默认构造函数,但这样会报错,所以不要这样。

2.3 lamda表达式

    auto comp = [](const int& lhs, const int& rhs) { return lhs > rhs; };
    priority_queue<int, vector<int>, decltype(comp)> q(comp);

官方文档中给出的具体的例子,https://en.cppreference.com/w/cpp/container/priority_queue/priority_queue:

复制代码

#include <iostream>
#include <queue>
#include <vector>
#include <utility>
 
using my_pair_t = std::pair<size_t,bool>;
 
using my_container_t = std::vector<my_pair_t>;
 
int main()
{
    auto my_comp =
        [](const my_pair_t& e1, const my_pair_t& e2) 
        { return e1.first > e2.first; };
    std::priority_queue<my_pair_t,
                        my_container_t,
                        decltype(my_comp)> queue(my_comp);
    queue.push(std::make_pair(5, true));
    ....
}

复制代码

但是结合第一节中的讨论,如果存储的类型是指针类型,那么参数就不能是const的,有两个具体的举例:

23. 合并K个升序链表
        auto cmp=[](const ListNode* & a,const ListNode* & b){return a->val>b->val;};//不能用const,是错的!OMG
        priority_queue<ListNode*,vector<ListNode*>,decltype(cmp)> pq(cmp);

注意,定义的cmp参数类型是指针类型,并且是const的,提交会报错:

 本人手动尝试了基本类型的指针:

复制代码

    auto comp = [](const int*& lhs, const int*& rhs) { return *lhs > *rhs; };
    priority_queue<int*, vector<int*>, decltype(comp)> q(comp);
    int* a=new int(5);
    int *b=new int(6);
    q.push(a);
    q.push(b);

复制代码

会报错:

error: cannot bind non-const lvalue reference of type 'const int*&' to an rvalue of type 'const int*'

搜索相关的结果都是  临时变量不能用作非const引用。所以这个错误的原因未知。

总的修改之后:

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

 ok.

复制代码

    auto comp = [](int* & lhs, int* & rhs) { return *lhs > *rhs; };
    priority_queue<int*, vector<int*>, decltype(comp)> q(comp);
    int* a=new int(5);
    int *b=new int(6);
    q.push(a);
    q.push(b);

复制代码

 ok.

所以如果T是指针类型,那么函数形参中不加const。

3.sort中的cmp函数与优先队列priority_queue的重载函数重载小于号)的区别

https://blog.csdn.net/weixin_44980441/article/details/107365028

复制代码

struct fruit
{
    string name;
    int price ;
    friend bool operator < (fruit a,fruit b)
    {
        return a.price>b.price;//低价优先!!!!。
    }
};

复制代码

bool cmp(fruit a,fruit b){
    return a.price>b.price;//高价优先
}

非常易错,到现在也分不清。。。

  • 8
    点赞
  • 23
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
priority_queue自定义比较可以通过重载小于操作符或者自定义函数对象来实现。 通过重载小于操作符,可以定义一个结构体或者类,并在其中重载小于操作符来比较元素的优先级。比如,可以定义一个结构体 MyType,重载小于操作符,使得元素按照 val 从大到小排序。 ``` struct MyType { int val; bool operator<(const MyType& other) const { return val > other.val; // 从大到小排序 } }; priority_queue<MyType> pq; ``` 另一种方法是通过定义函数对象来实现自定义比较。可以创建一个结构体或类,并在其中重载()操作符,然后作为模板参数传递给 priority_queue。比如,可以定义一个类 cmp,重载了()操作符,使得小的元素在队尾,优先级越低。 ``` struct cmp { bool operator()(int a,int b){ return a < b; // 小的放左边,即less } }; priority_queue<int,vector<int>,cmp> pq; ``` 还可以使用函数指针来自定义比较。可以定义一个函数,接受两个参数,并返回一个 bool 值,表示第一个元素是否比第二个元素优先。然后将函数指针作为模板参数传递给 priority_queue。例如,可以定义一个函数对象 cmp,重载了()操作符,使得大的元素在队尾,优先级越低。 ``` bool cmp(int a, int b) { return a > b; // 大的放左边,即greater } priority_queue<int, vector<int>, bool (*)(int, int)> pq(cmp); ``` 以上是三种常见的自定义比较方法,可以根据实际需求选择其中一种来实现自定义比较。<span class="em">1</span><span class="em">2</span><span class="em">3</span> #### 引用[.reference_title] - *1* [优先队列自定义比较函数](https://blog.csdn.net/yrk0556/article/details/87994924)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT0_1"}}] [.reference_item style="max-width: 33.333333333333336%"] - *2* [c++优先队列priority_queue自定义比较函数)](https://blog.csdn.net/qq_21539375/article/details/122128445)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT0_1"}}] [.reference_item style="max-width: 33.333333333333336%"] - *3* [一文看懂priority_queue自定义比较函数有几种方法](https://blog.csdn.net/weixin_36389889/article/details/129802998)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT0_1"}}] [.reference_item style="max-width: 33.333333333333336%"] [ .reference_list ]

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值