pair作为priority_queue元素时，自定义排序算法

最新推荐文章于 2024-03-28 18:54:51 发布

原创最新推荐文章于 2024-03-28 18:54:51 发布 · 4.4k 阅读

18 ·

CC 4.0 BY-SA版权

文章标签：

#数据结构

C++ 专栏收录该内容

7 篇文章

订阅专栏

本文介绍了C++中的priority_queue及其使用场景，重点讲解了如何通过pair结构配合less和greater函数实现自定义排序，并通过实例演示了降序和升序排列。同时，讨论了pair的基本概念和make_pair的用法，以及pair在优先队列中的默认排序规则。

priority_queue介绍

定义：priority_queue<Type, Container, Functional>

Type 就是数据类型。（int ，double等.)

Container 就是容器类型（Container必须是用数组实现的容器，默认vector，但不能用 list。）

默认降序排列，push()队尾插入，pop()弹出头元素。

priority_queue<int> a; //降序排列 等同于priority_queue<int,vector<int>,less<int>> b；
priority_queue<int,vector<int>,greater<int>> c; //升序排列

less表示数字大的优先级越大
greater表示数字小的优先级越大

pair 介绍

pair将两个数据组合成一组数据。
pair的实现是一个结构体，主要的两个成员变量是first、second。

pair<string,string> anon;
pair<string,int> word_count;
pair<string,int> name_age("Tom",18); //初始化

使用make_pair

pair<int,double> p1;
p1 = make_pair(1,1,2);

获取元素

p1.first;
p1.second;

pair比较大小

先按first比较，如果相等，再按照second比较。

默认排序

pair作为priority_queue元素时，使用默认less进行降序排列。

#include<iostream>
#include<string>
#include<functional>
#include<queue>
#include<vector>

using namespace std;
typedef pair<int,int> pairint;

int main()
{
        priority_queue<pairint,vector<pairint>,less<pairint> > test;
        pair<int,int> a(5,3);
        pair<int,int> b(1,10);
        pair<int,int> c(5,1);

        test.push(a);
        test.push(b);
        test.push(c);

        while (!test.empty()) {
                cout<<test.top().first<<","<<test.top().second<<endl;
                test.pop();
        }

        return 0;
}

输出：

5,3
5,1
1,10

当pair的first值相同时，比较second。

自定义排序顺序

通过自定义function object实现优先队列排序。

#include<iostream>
#include<string>
#include<functional>
#include<queue>
#include<vector>

using namespace std;
typedef pair<int,string> pairint;

class Comp{
        public:
        bool operator()(const pairint &n1,const pairint &n2)const
        {
                if(n1.first == n2.first)
                        return n1.first > n2.first;
                return n1.first > n2.first;
        }
};

int main()
{
        priority_queue<pairint,vector<pairint>,Comp > test;
        pair<int,string> a(5,"a");
        pair<int,string> b(1,"b");
        pair<int,string> c(5,"c");

        test.push(a);
        test.push(b);
        test.push(c);

        while (!test.empty()) {
                cout<<test.top().first<<","<<test.top().second<<endl;
                test.pop();
        }

        return 0;
}

输出：