[luogu 1631]序列合并

题目传送门

题意

给你两个长度都为 $n$ 的整数序列 $a_i$ 和 $b_i$ ，问你求出前$n$小的$a_i+b_j$

解题方法

我们一看到小，就首先想到把两个数组都从小到大排序一遍，然后我们发现了 $n$ 个如下的不等式
$$\begin{gathered} a_1+b_1\le a_1+b_2\le a_1+b_3\le ...\le a_1+b_n \\ {a}_2+b_1\le a_2+b_2\le a_2+b_3\le ...\le a_2+b_n \\ ... \\ {a}_n+b_1\le a_n+b_2\le a_n+b_3\le ...\le a_n+b_n \end{gathered}$$
然后我们就每次取出每个不等式的最小值然后丢进一个堆里面，每次取出堆顶然后把那个属于的不等式的最小值再扔回去。

代码

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
int a[100005],b[100005];
priority_queue<pair<int,int> >q;
int top[100005];
int main()
{
	int n;
	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]);
	sort(a+1,a+n+1);
	sort(b+1,b+n+1);
	for(int i=1;i<=n;i++)
		q.push(make_pair(-(a[i]+b[1]),i)),top[i]=1;
	for(int i=1;i<=n;i++)
	{
		int x,y;
		x=q.top().first;
		cout<<-x<<' ';
		y=q.top().second;
		q.pop();
		top[y]++;
		q.push(make_pair(-(a[y]+b[top[y]]),y));
	}
	return 0;
}