哈理工oj 最小的n个和【heap/最大堆的STL实现】

最新推荐文章于 2018-07-28 10:04:32 发布

mengxiang000000

最新推荐文章于 2018-07-28 10:04:32 发布

阅读量477

点赞数

CC 4.0 BY-SA版权

文章标签： hrbust 2062 哈理工oj

本文链接：https://blog.youkuaiyun.com/mengxiang000000/article/details/50771185

本文介绍了一种使用最大堆优化算法解决最小的n个和问题的方法，适用于给定两个数列，从每个数列中任意选取一个数进行相加，通过最大堆实现高效查找前n个最小和。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

最小的n个和

Time Limit: 1000 MS

Memory Limit: 32768 K

Total Submit: 145(44 users)

Total Accepted: 43(35 users)

Rating:

Special Judge: No

Description

给定A、B两个数列，各包含n个数，分别从A和B中任意取一个数相加得到和，这样会有n^2种结果(包括重复的)，求n^2个结果中前n个最小的和。

Input

有多组测试数据。
对于每组测试数据，第一行为n，第二行为数列A，第三行为数列B。
1<=n<=100000, 0 <= Ai, Bi <= 10^9。

Output

对于每组测试数据，输出一行，包含前n个最小的和，按照升序输出，两数之间用一个空格隔开。

Sample Input

5
1 3 4 2 0
7 3 5 2 11
10
74 50 47 45 38 64 19 2 84 69
91 46 44 7 67 1 40 60 78 41

Sample Output

2 3 3 4 4
3 9 20 26 39 42 43 45 46 46

Source

HCPC2014校赛训练赛 3

我们建立一个规模为n的最大堆，对于每一次加和，如果值比最大值小的话，我们把堆里边的跟节点（最大值）放在末尾，然后用加和值代替即可、

AC代码：

/*
    STL heap 最大堆
    make_heap(sum,sum+n);//建立一个规模为n的最大堆
    pop_heap(sum,sum+n);//把跟节点（最大值的数）放在最末尾、
    
*/

#include<stdio.h>
#include<algorithm>
#include<string.h>
using namespace std;
#define ll long long int
int a[100005],b[100005],sum[100005];
int main()
{
    int n;
    while(~scanf("%d",&n))
    {
        for(int i=0;i<n;i++)
            scanf("%d",&a[i]);
        sort(a,a+n);
        for(int i=0;i<n;i++)
            scanf("%d",&b[i]);
        sort(b,b+n);
        for(int i=0;i<n;i++)
            sum[i]=b[i]+a[0];
        make_heap(sum,sum+n);
        for(int i=1;i<n;i++)
        {
            for(int j=0;j<n;j++)
            {
                int tmp=a[i]+b[j];
                if(tmp>=sum[0])break;
                pop_heap(sum,sum+n);
                sum[n-1]=tmp;
                //push_heap(sum,sum+n);
            }
        }
        sort(sum,sum+n);
        for(int i=0;i<n-1;i++)
        {
            printf("%d ",sum[i]);
        }
        printf("%d\n",sum[n-1]);
    }
}