3555 Problem B 序列合并

问题 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;
}