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;
}