优先队列和结构体的运算符重载

自定义结构体是做题必须要学会的,而且经常需要用到自定义其排序规则。
优先队列也是,对于基本数据类型来说,不用自定义优先规则,但是对于自定义数据类型来说,确实是需要自己定义其内部的排序规则的。
今天发现原来两者的定义规则可以使用同一种方式来定义。

结构体规则定义

struct node{
    int x,y;
    node(){}
    node(int xx,int yy):x(xx),y(yy){}
    bool operator < (const node &b) const{
        if(x != b.x) return x > b.x;
        else return y > b.y;
    }
};

定义了结构体node,里面有两个变量x,y,并且定义了一个构造函数

node(int xx,int yy):x(xx),y(yy){}

这样可以在使用时,不用先申请在再使用了,可以直接使用。
其中包含了自定义排序规则的部分

    bool operator < (const node &b) const{//注意写法,一个都不能少
        if(x != b.x) return x > b.x;
        else return y > b.y;
    }

意为,先按照x从大到小排序,再按照y从大到小排序。
示例

#include<bits/stdc++.h>

using namespace std;

struct node{
    int x,y;
    node(){}
    node(int xx,int yy):x(xx),y(yy){}
    bool operator < (const node &b) const{
        if(x != b.x) return x > b.x;
        else return y > b.y;
    }
};

vector<node> ans;

int main() {

	srand((unsigned)time(nullptr));

	for(int i = 0;i < 5;i++) ans.push_back(node(rand()%6,rand()%10));

	sort(ans.begin(),ans.end());

	for(auto cur:ans) printf("%d %d\n",cur.x,cur.y);

	return 0;
}
//运行结果
4 1
3 6
2 9
1 5
1 1

可以发现,排序结果是按照x从大到小,x相同时,按照y从大到小进行排序的。
那么既然排序规则定义在了里面,如果还想让数据按照别的顺序进行排序怎么办?要在结构体里面再写一个运算符重载吗?其实,虽然已经在结构体内部写好了排序规则,这样在排序时,不用再添加第三个自定义的排序规则了,但是其实还是可以再在结构体外部写新的排序规则。
示例

#include<bits/stdc++.h>

using namespace std;

struct node{
    int x,y;
    node(){}
    node(int xx,int yy):x(xx),y(yy){}
    bool operator < (const node &b) const{
        if(x != b.x) return x > b.x;
        else return y > b.y;
    }
};

vector<node> ans;

//在结构体外部定义了新的排序规则
//先是按照x从小到大,若x相同,按照y从小到大
bool cmp(node &a,node &b){
    if(a.x != b.x) return a.x < b.x;
    else return a.y < b.y;
}

int main() {

	srand((unsigned)time(nullptr));

	for(int i = 0;i < 5;i++) ans.push_back(node(rand()%6,rand()%10));

	sort(ans.begin(),ans.end(),cmp);

	for(auto cur:ans) printf("%d %d\n",cur.x,cur.y);

	return 0;
}
//运行结果
1 5
3 0
3 4
5 3
5 7

可以看到,这次虽然在结构体里面已经定义了一个排序规则了,又在结构体外面定义了规则,在使用sort进行数组的排序时,是按照结构体外面的cmp定义的规则进行排序的。是可以的

优先队列优先级定义

#include<bits/stdc++.h>

using namespace std;

struct node{
    int x,y;
    node(){}
    node(int xx,int yy):x(xx),y(yy){}
    bool operator < (const node &b) const{
        if(x != b.x) return x > b.x;
        else return y > b.y;
    }
};

priority_queue<node> q;

int main() {

	srand((unsigned)time(nullptr));

	for(int i = 0;i < 10;i++)
        q.push(node(rand()%10,rand()%6));
    while(q.size()){
        printf("%d %d\n",q.top().x,q.top().y);
        q.pop();
    }
	return 0;
}
//运行结果
0 8
1 4
3 6
4 0
4 6

可以看到,其在结构体内的优先级定义方法和上述的用于结构体数组排序时的方法是一致的。
之前我在使用优先队列的优先级定义时,使用的是下述代码

struct node{
    int x,y;
    node(){}
    node(int xx,int yy):x(xx),y(yy){}

    friend bool operator < (const node &a,const node &b){
        if(a.x != b.x) return a.x > b.x;
        else return a.y > b.y;
    }
}

现在可以统一起来,只用使用第一个就好了。

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值