序列合并【洛谷P1631】

题目描述

有两个长度都是N的序列A和B，在A和B中各取一个数相加可以得到N²个和，求这N²个和中最小的N个。

输入格式

第一行一个正整数N；
第二行N个整数Ai,满足Ai<=Ai+1且Ai<=10^9;
第三行N个整数Bi, 满足Bi<=Bi+1且Bi<=10^9.

输出格式

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

样例输入

3
2 6 6
1 4 8

样例输出

3 6 7

思路
先将b数组排序，然后将a数组的每个数a1,a2,…an分别加入每个b[1]中，使用priority_queue进行a[i]+b[1]值的维护（小根堆，最小值为根），每次弹出根的i和j,将a[i]再次加到b[j+1]上，加入堆中，时间复杂度O(nlogn)

c++ code:

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=100005;
struct node
{
	int sum;
	int ii;
	int jj;
}tt,xx,yy;
bool operator<(const node &a,const node &b)
{
	return a.sum>b.sum;
}
priority_queue<node>q;
int ans[N];
int a[N],b[N];
int main()
{
	//freopen("1631.in","r",stdin);
	//freopen("1631.out","w",stdout);
	int n,i,j,num=1;
	scanf("%d",&n);
	for(i=1;i<=n;i++)
		scanf("%d",&a[i]);
	for(i=1;i<=n;i++)
		scanf("%d",&b[i]);
	sort(b+1,b+1+n);
	for(i=1;i<=n;i++)
	{
		tt.sum=a[i]+b[1];
		tt.ii=i;
		tt.jj=1;
		q.push(tt);
	}
	while(num<=n)
	{
		xx=q.top();
		q.pop(); 
		ans[num]=xx.sum;
		yy.ii=xx.ii;
		yy.jj=xx.jj+1;
		yy.sum=a[yy.ii]+b[yy.jj];
		q.push(yy);
		num++;
	}
	for(i=1;i<=n;i++)
		printf("%d ",ans[i]);
	return 0;
}