2018年第三阶段个人训练赛第十场2654 Problem M 序列合并

题目描述

有两个长度都是N的序列A和B，在A和B中各取一个数相加可以得到N^2个和，求这N^2个和中最小的N个。

 

输入

第一行一个正整数N；

第二行N个整数Ai,满足Ai<=Ai+1且Ai<=10^9;

第三行N个整数Bi, 满足Bi<=Bi+1且Bi<=10^9.

 

输出

仅一行，包含N个整数，从小到大输出这N个最小的和，相邻数字之间用空格隔开。

 

样例输入

3
2 6 6
1 4 8

 

样例输出

3 6 7 

 

提示

对于50%的数据中，满足1<=N<=1000；
对于100%的数据中，满足1<=N<=100000。

思路：用优先队列

先将a[1]与b[i]相加入队列，并标记a[1]与b数列中相加数的下标，并记录sum[i]=1,代表a的下标，题意已说明a,b数列按从小到大排序，所以a[1]+b[1]肯定最小，最先出队列，然后sum[1]++，所以sum[1]=2，所以a[2]+b[1]入队列，并标记b的下标即1；再次输出队列中最小的数，其标记的下标为i，然后sum[i]++，所以然后入队列的是a[sum[i]]+b[i]，并标记b的下标即i。输出n个数为止。

也就是先将a[1]与b[i]相加入队列，然后a[i]+b[i]出队列，随之a[i+1]+b[i]入队列，优先队列自动排序，每次输出最小，即为题目所求。

#include<cstdio>
#include<cstring>
#include<math.h>
#include<algorithm>
#include<queue>
#include<vector>
#include<iostream>
#include<map>
#include<queue>
#define mes(a,b) memset(a,b,sizeof(a))
#define rep(i,m,n) for(i=m;i<=n;i++)
typedef long long ll;
using namespace std;
int max3(int a,int b,int c){return max(max(a,b),c);}
ll min3(ll a,ll b,ll c){return min(min(a,b),c);}
const double PI=acos(-1);
const int inf=0x3f3f3f3f;
const double esp=1e-6;
const int maxn=1e6+5;
const int mod=1e9+7;
ll gcd(ll a,ll b){return b?gcd(b,a%b):a;}
ll lcm(ll a,ll b){return a/gcd(a,b)*b;}
ll inv(ll b){if(b==1)return 1; return (mod-mod/b)*inv(mod%b)%mod;}
ll fpow(ll n,ll k){ll r=1;for(;k;k>>=1){if(k&1)r=r*n%mod;n=n*n%mod;}return r;}
ll Fpow(ll n,ll k){ll r=1;for(;k;k>>=1){if(k&1)r=r*n;n=n*n;}return r;}
long long  a[100010],b[100010],sum[100010]={0};
priority_queue<pair<int,int>,vector<pair<int,int> >,greater<pair<int,int> > > q;
int main()
{
   int n;
   scanf("%d",&n);
   int i;
   for(i=1;i<=n;i++)
   {
       scanf("%lld",&a[i]);
   }
   for(i=1;i<=n;i++)
   {
       scanf("%lld",&b[i]);
       sum[i]=1;
       q.push(pair<int,int>(a[1]+b[i],i));
   }
   int z;
   for(int i=1;i<=n;i++)
	{
		printf("%d ",q.top().first);
		z=q.top().second;
		q.pop();
		sum[z]++;
		q.push(pair<int,int>(a[sum[z]]+b[z],z));
	}
   return 0;

}