洛谷 P1631 序列合并

题目：序列合并

思路：
摘自@Red_w1nE的洛谷博客

首先，把A和B两个序列分别从小到大排序，变成两个有序队列。
这样，从A和B中各任取一个数相加得到N2个和，可以把这些和看成形成了n个有序表/队列：
A[1]+B[1] <= A[1]+B[2] <= … <= A[1]+B[N]
A[2]+B[1] <= A[2]+B[2] <= … <= A[2]+B[N]
……
A[N]+B[1] <= A[N]+B[2] <= … <= A[N]+B[N]
接下来，就相当于要将这N个有序队列进行合并排序：
首先，将这N个队列中的第一个元素放入一个堆中；
然后；每次取出堆中的最小值。若这个最小值来自于第k个队列，那么，就将第k个队列的下一个元素放入堆中。
时间复杂度：O(NlogN)。

代码：

#include<bits/stdc++.h>
using namespace std;

#define maxn 100000
#define read(x) scanf("%d",&x)
#define ll long long

struct Pair {
	int x,y;
	Pair(){}
	Pair(int xx,int yy) {x=xx,y=yy;}
	bool operator < (const Pair& oth) const {
		return x>oth.x;
	}
};

int n;
int a[maxn+5],b[maxn+5];
priority_queue<Pair> que;
int t[maxn+5];

int main() {
	read(n);
	for(int i=1;i<=n;i++) read(a[i]);
	for(int i=1;i<=n;i++) read(b[i]);
	
	for(int i=1;i<=n;i++) {
		que.push(Pair(a[1]+b[i],i));
		t[i]=1;
	}
	
	for(int i=1;i<=n;i++) {
		Pair h=que.top();que.pop();
		if(t[h.y]!=n) {
			t[h.y]++;
			que.push(Pair(a[t[h.y]]+b[h.y],h.y));
		}
		printf("%d ",h.x);
	}
	
	return 0;
}