题目
题目描述 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);
}
}