问题描述:
有两个长度都是
的序列
和
,在A和B中各取一个数相加可以得到
个和,求在这
和中最小的
个。
入输出格式
输入格式:
第一行一个正整数N;
第二行 个整数 满足 且
第三行 个整数 满足 且
【数据规模】
对于 的数据中,满足 ;
对于 的数据中,满足
输出格式:
输出仅一行,包含 个整数,从小到大输出这 个最小的和,相邻数字之间用空格隔开。
输入输出样例
输入
3
2 6 6
1 4 8
输出
3 6 7
解:
我们发现这道题两个序列
是单调不下降的,所以我们立即注意到:
…
所以我们只需要
个指针指向这
行未用过的值放入堆中。
每次取堆顶元素把这个指针对应的值输出并且把指针向后移即可
堆的每次操作都是
的,要执行
次所以总时间复杂度为
#include<bits/stdc++.h>
using namespace std;
inline void rin(int &t)
{
int k=1;
t=0;
char c=getchar();
while(!isdigit(c)){if(c=='-')k=-1;c=getchar();}
while(isdigit(c)){t=t*10+c-'0';c=getchar();}
t*=k;
}
const int maxn=100000+5;
int a[maxn],b[maxn];
typedef pair<int,int> P;
#define x first
#define y second
int n;
priority_queue<P,vector<P>,greater<P> > q;
int c[maxn];
int main()
{
rin(n);
for(int i=1;i<=n;i++)rin(a[i]);
for(int i=1;i<=n;i++)rin(b[i]),q.push(make_pair(a[1]+b[i],i)),c[i]=1;
P tmp;
for(int i=1;i<=n;i++)
{
tmp=q.top();
q.pop();
printf("%d ",tmp.x);
q.push(make_pair(a[++c[tmp.y]]+b[tmp.y],tmp.y));
}
return 0;
}