洛谷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;
}