codevs1245 最小的N个和-白红宇

codevs1245 最小的N个和

阅读量：5230 次

发布时间：2019-06-14

本文共 1332 字，大约阅读时间需要 4 分钟。

1245

题目描述

Description

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

输入描述

Input Description

第一行输入一个正整数N；第二行N个整数Ai 且Ai≤10^9；第三行N个整数Bi，

且Bi≤10^9

输出描述

Output Description

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

空格隔开。

样例输入

Sample Input

1 3 2 4 5

6 3 4 1 7

样例输出

Sample Output

2 3 4 4 5

数据范围及提示

Data Size & Hint

【数据规模】对于 100%的数据，满足 1≤N≤100000。

思路

先把两个数组从小到大排个序

先求出a数组中每个数与b数组中第一个数的和，作为堆的初始值，建一个小根堆，以num为优先级，y其次

由于我们不确定是a中每个数加b中最小数比较小，还是b中每个数加a中最小数比较小

所以要让堆动起来，插入的元素就是a中当时的元素与b中下一个元素的和，每次弹出堆顶元素，重复n次即可

#include
     
      #include
      
       #include
       
        using namespace std;int n,a[100001],b[100001];struct node{    int y,num;    bool operator <(const node &v)const    {        return num>v.num;    }};node k;priority_queue
        
         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<
         
          <<' ';    }    return 0;}