关于自定义排序是一些注意的地方

对于这种情况:


#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 (有部分是参考这里面的)

猜你喜欢

转载自blog.csdn.net/weixin_42373330/article/details/81540117