priority_queue

最新推荐文章于 2025-05-23 19:34:35 发布

转载最新推荐文章于 2025-05-23 19:34:35 发布 · 61 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/zuiaimiusi/p/10708467.html

本文介绍如何使用C++ STL中的priority_queue(优先队列)解决寻找两个有序序列组合后的最小N个和的问题。通过自定义结构体并重载比较运算符，实现了对结构体元素的正确排序，进而利用优先队列高效地找到所需结果。

priority_queue本质是一个堆。

1. 头文件是#include<queue>

2. 关于priority_queue中元素的比较

　　模板申明带3个参数：priority_queue<Type, Container, Functional>，其中Type 为数据类型，Container为保存数据的容器，Functional 为元素比较方式。

　　Container必须是用数组实现的容器，比如vector,deque等等，但不能用 list。STL里面默认用的是vector。

3. priority_queue<int> q; （这个int就是你要申请的类型，此时默认的为大根堆，即依次弹出的数的顺序为从大到小）
     如果要从小到大，则是这样：
     priority_queue<int, vector<int>, greater<int> > q;
    所以结构体就很容易想到了啊:
          priority_queue<lpl> (lpl 为结构体的类型)
但是由于是结构体，那么它的比较大小的方式则也需要你来定义，所以你需要重载运算符。

题目描述

有两个长度都是N的序列A和B，在A和B中各取一个数相加可以得到N^2个和，求这N^2个和中最小的N个。

输入输出格式
输入格式：

第一行一个正整数N；

第二行N个整数Ai,满足Ai<=Ai+1且Ai<=10^9;

第三行N个整数Bi, 满足Bi<=Bi+1且Bi<=10^9.

【数据规模】

对于50%的数据中，满足1<=N<=1000；

对于100%的数据中，满足1<=N<=100000。
输出格式：

输出仅一行，包含N个整数，从小到大输出这N个最小的和，相邻数字之间用空格隔开。

输入输出样例

3
2 6 6
1 4 8
3 6 7

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 const int maxn = 1e5 + 5;
 4 struct lpl
 5 {
 6     int a;
 7     int b;
 8     int sum;
 9 }lin;
10 bool operator < (lpl a, lpl b)
11 {
12 
13     if(a.sum > b.sum)    return true;
14     return false;
15 }
16 priority_queue<lpl> q;
17 int n;
18 int a[maxn], b[maxn];
19 void putit()
20 {
21     scanf("%d", &n);
22     for(int i = 1; i <= n; ++i)
23         scanf("%d", &a[i]);
24     for(int i = 1; i <= n; ++i)
25         scanf("%d", &b[i]);
26     for(int i = 1; i <= n; ++i)
27         lin.a= 1, lin.b= i, lin.sum = a[1] + b[i], q.push(lin);
28 }
29 void workk()
30 {
31     for(int i = 1; i <= n; ++i)
32     {
33         lpl now = q.top();
34         q.pop();
35         printf("%d ", now.sum);
36         now.sum = a[now.a+ 1] + b[now.b];
37         now.a++;
38         q.push(now);
39     }
40 }
41 int main()
42 {
43     putit();
44     workk();
45     return 0;
46 }

转载于:https://www.cnblogs.com/zuiaimiusi/p/10708467.html