最小的N個和(優先佇列)
阿新 • • 發佈:2018-12-24
題目描述
有兩個長度為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;
}