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

解题思路

因为两个序列都是不下降的有序序列，所以满足以下不等式 

a1+b1<=a1+b2<=a1+b3<=...<=a1+bn

a2+b1<=a2+b2<=a2+b3<=...<=a2+bn

...

an+b1<=an+b2<=an+b3<=...<=an+bn

所以，这n^n个和里的最小值在a1+b1,a2+b1...an+b1里（这时最小的当然是a1+b1），然后去掉最小的a1+b1,剩下的不等式变为以下

a1+b2<=a1+b3<=...<=a1+bn

a2+b1<=a2+b2<=a2+b3<=...<=a2+bn

...

an+b1<=an+b2<=an+b3<=...<=an+bn

最小值还是第一列的那n个和里最小的。所以始终用优先队列维护n个和，直到得到最小的N个和。

代码如下

#include <iostream>
#include <queue>
#define maxn 100005
using namespace std;
int a[maxn], b[maxn];
int t[maxn];
struct mes{
	int x, sum;
	mes(int x, int s): x(x), sum(s){	}
	bool operator<(const mes& a)const{
		return sum > a.sum;
	}
};
int main()
{
	int n;
	while(cin >> n){
		for(int i = 1; i <= n; i ++)
			scanf("%d", &a[i]);
		for(int i = 1; i <= n; i ++)
			scanf("%d", &b[i]);
		priority_queue<mes> que;
		for(int i = 1; i <= n; i ++){
			que.push(mes(i, a[i] + b[1]));
			t[i] = 1;
		}		
		int cnt = 0;
		bool first = true;
		while(cnt < n){
			mes top = que.top();
			que.pop();
			if(first)
				first = false;
			else
				printf(" ");
			printf("%d", top.sum);
			t[top.x] ++;
			que.push(mes(top.x, a[top.x] + b[t[top.x]]));
			cnt ++;
		}
		printf("\n");
	}
	return 0;
}