题目描述

　　问题：序列和的前n小元素
　　给出两个长度为n的有序表A和B, 在A和B中各任取一个, 可以得到 n^2 个和. 求这些和最小的n个。

输入输出格式

输入格式：

　　输入数据共三行。
　　第一行，一个整数值n （ n <= 10^4 ）。
　　第二，第三行，各有n个从小到大排好序的整数，每个整数间有一个空格间隔。（每个整数均小于2^30）

输出格式：

　　输出数据一行,这些和最小的n个数，从小到大输出，每个整数之间一个空格间隔。

输入输出样例

输入样例#1：

10
158443636 284841496 317570810 452077938 476117840 485745865 646752262 998776724 1022863800 1038269864 
8345019 96770572 114185139 211677750 365253930 495542735 670357622 765440815 783523273 835082336 

输出样例#1：

166788655 255214208 272628775 293186515 325915829 370121386 381612068 399026635 414341382 431755949

提示信息

数据范围：
　　30%数据，n <= 10^2。
　　50%数据，n <= 10^3。
　　100%数据，n <= 10^4。

算法分析：

二叉堆模板题：用优先队列实现

思路：

暴力枚举每个数，使堆内始终有n个数。

题中有“两个长度为n的有序表A和B”

故，在第二层循环时，若已有n个数，则只需判断一次即可退出循环。

#include<bits/stdc++.h>
using namespace std;
priority_queue<int,vector<int> > p;
int a[1000010];
int b[1000010];
int c[1000010];
int n,len;
int main()
{cin>>n;for(int i=1;i<=n;i++)cin>>a[i];for(int i=1;i<=n;i++)cin>>b[i];for(int i=1;i<=n;i++){for(int j=1;j<=n;j++){if(len<n){p.push(a[i]+b[j]);len++;}//先把堆填充满n个数else{if(a[i]+b[j]<=p.top()){p.pop();p.push(a[i]+b[j]);}//优先要小的，比堆头小就扔掉堆头else break;//如果a[i]+b[j]小于堆头,那么对于有序表b来说，a[i]+b[j+1]必定大于堆头}}}len=0;while(!p.empty()){c[++len]=p.top();p.pop();}for(int i=1;i<=n;i++){cout<<c[n-i+1]<<' ';}return 0;
}