最小的N个和(优先队列)
题目描述
有两个长度为N的序列A和B,在A和B中各任取一个数相加可以得到N2个和,求这N2个和中最小的N个。
输入
第一行输入一个正整数N(1<=N<=100000);
第二行N个整数Ai且Ai<=109;第三行N个整数Bi且Bi<=109。
输出
输出仅一行,包含n个整数,从小到大输出这n个最小的和,相邻数字之间用空格隔开。
样例输入
5
1 3 2 4 5
6 3 4 1 7
样例输出
2 3 4 4 5
解析:用优先队列,每一个A数组里面的值都有一个值,当这个值被用了之后就转换成这个值和B数组里面的下一个值求和就好了
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10
int n,a[N],b[N];
struct node
{
int y,num;
bool operator <(const node &v)const
{
return num>v.num;
}
};
node k;
priority_queue<node>q;
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
cin>>a[i];
for(int i=1;i<=n;i++)
cin>>b[i];
sort(a+1,a+n+1);
sort(b+1,b+n+1);
for(int i=1;i<=n;i++)
{
k.y=1;k.num=a[i]+b[1];
q.push(k);
}
int s=1;
while(s<=n)
{
node now=q.top();
q.pop();
if(now.y+1<=n)
{
k.y=now.y+1;
k.num=now.num-b[now.y]+b[now.y+1];
q.push(k);
}
s++;
cout<<now.num<<' ';
}
return 0;
}