问题 B: 序列合并
时间限制: 1 Sec 内存限制: 128 MB
提交: 88 解决: 33
题目描述
有两个长度都为N的序列A和B,在A和B中各取一个数相加可以得到N2个和,求这N2个和中最小的N个。
输入
第一行一个正整数N(1 <= N <= 100000)。 第二行N个整数Ai,满足Ai <= Ai+1且Ai <= 109 第三行N个整数Bi,满足Bi <= Bi+1且Bi <= 109
输出
输出仅有一行,包含N个整数,从小到大输出这N个最小的和,相邻数字之间用空格隔开。
样例输入
3 2 6 6 1 4 8
样例输出
3 6 7
提示
建议用最小堆实现。
经验总结
这题。。。还是要用堆实现,这个算法。。。我是木有想到的= = 借鉴大神的代码思想,思想描述如下
因为两个数组都是有序递增,所以先建立堆结构(大端堆),将其中一个数组A的第一个元素与另一个数组B的所有元素相加的结果依次放入,然后再从A的第二个元素开始,依次与B中所有元素相加,相加结果与堆中根结点比较,如果比堆的根结点小,删除根结点,将此结果加入,如果比根结点大,那么跳出此次循环,访问A的下一个元素。
此时堆为大端堆,用数组将其倒置后输出即可。 学到了~~( •́ .̫ •̀ )
正确代码
#include <cstdio>
#include <queue>
using namespace std;
const int maxn=100010;
int A[maxn],B[maxn],ans[maxn],n;
priority_queue<int> q;
int main()
{
int data;
while(~scanf("%d",&n))
{
for(int i=1;i<=n;++i)
{
scanf("%d",&A[i]);
}
for(int i=1;i<=n;++i)
{
scanf("%d",&B[i]);
q.push(B[i]+A[1]);
}
for(int i=2;i<=n;++i)
{
for(int j=1;j<=n;++j)
{
if(A[i]+B[j]<q.top())
{
q.pop();
q.push(A[i]+B[j]);
}
else
break;
}
}
for(int i=n-1;i>=0;--i)
{
ans[i]=q.top();
q.pop();
}
for(int i=0;i<n;++i)
{
printf("%d",ans[i]);
if(i!=n-1)
printf(" ");
}
printf("\n");
}
return 0;
}