【题解】洛谷P1631 序列合并

https://www.luogu.org/blog/user23845/solution-p1631

这里我把序号存在结构体里，这样堆里需要比较结构体，重载运算符。

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
using namespace std;
const int maxn=100010;
int n; 
int a[maxn],b[maxn];
struct R
{
	int ai,bi;
};
bool operator < (const R &x1,const R &x2)
{
	return a[x1.ai]+b[x1.bi]>a[x2.ai]+b[x2.bi];	
} 
priority_queue<R> q;
int main()
{
	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]);
	}
	for(int i=1;i<=n;i++)
	{
		q.push((R){i,1});
	}
	for(int i=1;i<=n;i++)
	{
		R tmp=q.top();
		q.pop();
		printf("%d ",a[tmp.ai]+b[tmp.bi]);
		tmp.bi++;
		q.push(tmp);
	}
	return 0;
}