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

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

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

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;
}

输出:

1,b
5,a
5,c
### C++ 中 `priority_queue` 的使用场景及原因 #### 1. **优先级队列的核心功能** `priority_queue` 是一种基于堆的数据结构,能够高效地维护一组具有优先级的元素集合。它通过二叉堆(默认最大堆)来实现快速访问最高优先级的元素[^1]。 #### 2. **为什么选择 `priority_queue`?** - **高效的插入与删除操作** 插入 (`push`) 和删除 (`pop`) 操作的间复杂度均为 \(O(\log n)\),这使得 `priority_queue` 非常适合处理动态变化的大规模数据集[^3]。 - **自动排序能力** `priority_queue` 自动将新加入的元素放置到合适位置,从而始终保持顶部元素是最优解(即最大值或最小值)。这种特性非常适合解决涉及最优选择的问题[^2]。 - **灵活性高** 支持自定义比较函数,可以轻松调整为小顶堆或其他特定需求的优先级逻辑[^3]。 #### 3. **适用场景分析** 以下是常见的应用场景: ##### (1)**贪心算法中的最优选择问题** 当需要不断选取当前状态下最优的选择,`priority_queue` 可以提供最高效的解决方案。例如: - 背包问题变种:每次取价值最高的物品放入背包。 - 合并K个有序链表:利用小顶堆找到每轮中最小节点。 ```cpp // 示例代码:合并 K 个升序链表 struct ListNode { int val; ListNode* next; }; bool operator<(const ListNode* a, const ListNode* b) { return a->val > b->val; // 小顶堆 } ListNode* mergeKLists(vector<ListNode*>& lists) { priority_queue<ListNode*> pq; for(auto& list : lists){ if(list != nullptr) pq.push(list); } ListNode dummy(0), *tail = &dummy; while(!pq.empty()){ auto node = pq.top(); pq.pop(); tail->next = node; tail = tail->next; if(node->next != nullptr){ pq.push(node->next); } } return dummy.next; } ``` ##### (2)**Dijkstra 或 Prim 算法中的优化** 在图论中最短路径或最小生成树计算过程中,通常会用到优先级队列加速更新距离的过程。比如 Dijkstra 算法可以通过 `priority_queue` 实现更高效的松弛操作。 ```cpp // 示例代码:Dijkstra 最短路算法 void dijkstra(int start, vector<vector<pair<int,int>>>& graph, vector<int>& dists){ priority_queue<pair<int,int>, vector<pair<int,int>>, greater<>> pq; pq.emplace(0,start); dists[start]=0; while(!pq.empty()){ auto [d,u]=pq.top(); pq.pop(); if(d>dists[u]) continue; for(const auto &[v,w]:graph[u]){ if(d+w<dists[v]){ dists[v]=d+w; pq.emplace(d+w,v); } } } } ``` ##### (3)**Top-K 类型问题** 如果目标是从大量无序数据集中找出前 K 大或者前 K 小的元素,则可以直接借助 `priority_queue` 来完成筛选工作[^3]。 ```cpp // 找出数组中最大的三个数 vector<int> findTopKElements(const vector<int>& nums, int k){ priority_queue<int, vector<int>, greater<int>> minHeap(nums.begin(),nums.begin()+k); for(size_t i=k;i<nums.size();++i){ if(minHeap.top()<nums[i]){ minHeap.pop(); minHeap.push(nums[i]); } } vector<int> res; while(!minHeap.empty()){ res.push_back(minHeap.top()); minHeap.pop(); } reverse(res.begin(),res.end()); return res; } ``` #### 4. **总结** `priority_queue` 提供了一种简单而强大的工具用于管理带权值的对象序列,并能在多种实际应用场合发挥重要作用。无论是寻找局部极值还是全局规划决策过程,都离不开它的身影。 ---
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

硬码农二毛哥

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值