Codeup堆：序列合并

题目描述

有两个长度都为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

---

#include<bits/stdc++.h>
using namespace std;
const int maxn=100010;
int a[maxn],b[maxn],ans[maxn],n;
priority_queue<int>q;//即从大到小的队列 
int main(){
	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;
}