对于这种情况:
#include<iostream>
#include <queue>
#include <algorithm>
using namespace std;
typedef pair<int,int> p;
bool cmp(const p &a,const p &b)
{
if(a.first==b.first)
return a.second>b.second;//当a 相等时,b按从大到小排序---注意一下这与下面的两个案例相反
return a.first<b.first;// a 按从小到大排序---注意一下这与下面的两个案例相反
}
p a[100];
int main()
{
a[1]=make_pair(1,2);
a[2]=make_pair(1,3);
a[3]=make_pair(2,3);
a[4]=make_pair(2,2);
a[5]=make_pair(1,8);
sort(a+1,a+6,cmp);//这里的a+1与a+6分别指向的是a[1]这个位置和a[5]这个位置,然后将其排序
for(int i=1;i<=5;i++)
cout<<a[i].first<<" "<<a[i].second<<endl;
return 0;
}
输出结果:
而对于像这样的情况:
#include<iostream>
#include <queue>
using namespace std;
typedef pair<int,int> p;
bool operator<(p a,p b)
{
if(a.first==b.first)
return a.second>b.second;//当a 相等时,b按从小到大排序---注意一下
return a.first<b.first;// a 按从大到小排序---注意一下
}
int main()
{
priority_queue<p> q1;
q1.push(make_pair(1,2));
q1.push(make_pair(1,3));
q1.push(make_pair(2,3));
q1.push(make_pair(2,2));
q1.push(make_pair(1,8));
while(!q1.empty())
{
cout<<(q1.top()).first<<" "<<(q1.top()).second<<endl;
q1.pop();
}
return 0;
}
输出结果:
还可以这样写,结果与上面那个一样:
可以自定义一个比较类,Compare
priority_queue中的三个参数,后两个可以省去,因为有默认参数,不过如果,有第三个参数的话,必定要写第二个参数。
而且这个是一个类,这个类里重载operator(),和自定义sort排序不同,sort只需bool cmp(……)(当然sort也可以弄一个比较类,再重载operator()),若是priority_queue中写为sort的cmp形式则报错,如:bool cmp1(const node &a, const node&b)//报错!
代码:
#include<iostream>
#include <queue>
using namespace std;
typedef pair<int,int> p;
struct cmp
{
bool operator()(const p &a,const p &b)
{
if(a.first==b.first)
return a.second>b.second;//当a 相等时,b还是按从小到大排序---注意一下
return a.first<b.first;// a 还是按从大到小排序---注意一下
}
};
int main()
{
priority_queue<p,vector<p>, cmp> q1;
q1.push(make_pair(1,2));
q1.push(make_pair(1,3));
q1.push(make_pair(2,3));
q1.push(make_pair(2,2));
q1.push(make_pair(1,8));
while(!q1.empty())
{
cout<<(q1.top()).first<<" "<<(q1.top()).second<<endl;
q1.pop();
}
return 0;
}
输出结果:
关于更多优先队列priority_queue自定义比较函数用法整理
可阅读这篇文章:https://blog.csdn.net/bat67/article/details/77585312 (有部分是参考这里面的)