最小的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;
}

转载于:https://www.cnblogs.com/ffgcc/p/10546435.html