序列合并(codevs1245 最小的N个和)

来源:互联网 发布:三星电视上不了网络 编辑:程序博客网 时间:2024/06/07 15:43

传送门

题目

题目描述 Description
有两个长度为 N 的序列 A 和 B,在 A 和 B 中各任取一个数可以得到 N^2 个和,求这N^2 个和中最小的 N个。

输入描述 Input Description
第一行输入一个正整数N;第二行N个整数Ai 且Ai≤10^9;第三行N个整数Bi,
且Bi≤10^9

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

过程

此题为上午考试的第二题。考试的时候自己写了一个奇怪的程序,用了各种奇技淫巧来减小时间复杂度,结果得了60分。
考试结束以后,和别人一起讨论这道题,原来用堆实现是如此简单!由于C++的STL有自带的堆,于是我上网查了一个STL堆的模板,照着敲了一份代码,却一直编译失败(555)。将模板复制下来比较,好像没什么不对的。于是我向xym学长请教了小根堆的用法,来来回回问了好几次,在此表示感谢~
ps:用C++的STL实现大根堆和小根堆时,大根堆重载小于号,小根堆重载大于号;实现小根堆时,两个”>”之间要留有空格,否则会编译失败。
代码中我写了两种重载小于号的方法,都是可以的。
代码:

#include<queue>#include<cstdio>#include<algorithm>using namespace std;int n,i;int a[100001];int b[100001];struct aa{    int p,sum;//  bool operator >(const aa other)const//  {//      return sum>other.sum;//  }};bool operator >(aa a,aa b){    return a.sum>b.sum;}aa t;priority_queue<aa,vector<aa>,greater<aa> >q;void read(int &n){    n=0;    char c=getchar();    while(c<'0'||c>'9') c=getchar();    while(c>='0'&&c<='9') n=n*10+c-48,c=getchar();}main(){    freopen("sequence.in","r",stdin);    freopen("sequence.out","w",stdout);    read(n);    for(i=1;i<=n;i++)      read(a[i]);    for(i=1;i<=n;i++)      read(b[i]);    sort(a+1,a+n+1);    sort(b+1,b+n+1);    for(i=1;i<=n;i++)    {        t.p=1;        t.sum=b[1]+a[i];        q.push(t);    }    for(i=1;i<=n;i++)    {        t=q.top();        q.pop();        printf("%d ",t.sum);        t.sum=t.sum-b[t.p]+b[++t.p];        q.push(t);    }}
0 0