洛谷p1631 序列合并

题目描述

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

输入输出格式

输入格式：

第一行一个正整数N；

第二行N个整数 A_iAi​ , 满足 A_i\le A_{i+1}Ai​≤Ai+1​ 且 A_i\le 10^9Ai​≤109 ;

第三行N个整数 B_iBi​ , 满足 B_i\le B_{i+1}Bi​≤Bi+1​ 且 B_i\le 10^9Bi​≤109 .

【数据规模】

对于50%的数据中，满足1<=N<=1000；

对于100%的数据中，满足1<=N<=100000。

输出格式：

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

输入输出样例

输入样例#1：

3
2 6 6
1 4 8

输出样例#1： 

3 6 7

#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
using namespace std;
struct node{
	int x,y,sum;
	friend bool operator <(node a,node b)
	{
		return a.sum>b.sum;
	}
};
int n;
int a[100005],b[1000055];
int main(){
	priority_queue<node> q;
	cin>>n;
	node wzx;
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&a[i]);
	}
	for(int i=1;i<=n;i++)
	{
		scanf("%d",&b[i]);
	}
	for(int i=1;i<=n;i++)
	{
		wzx.x=i;
		wzx.y=1;
		wzx.sum=a[wzx.x]+b[wzx.y];
		q.push(wzx);
	}
	for(int i=1;i<=n;i++)
	{
		wzx=q.top();
		q.pop();
		printf("%d ",wzx.sum);
		wzx.y++;
		wzx.sum=a[wzx.x]+b[wzx.y];
		q.push(wzx);
	}
	return 0;
}