首页 > 代码库 > wikioi 1245 小顶堆

wikioi 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

5

1 3 2 4 5 
6 3 4 1 7

样例输出 Sample Output

2 3 4 4 5

数据范围及提示 Data Size & Hint

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


这题刘汝佳白书中190页里有,挺好的堆思想题。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>
#include<queue>
#include<set>
#include<bitset>
#define INF 100007
using namespace std;
typedef unsigned long long ull;
struct abc
{
    int sum,i;
    abc(int sum,int i):sum(sum),i(i){}
    bool operator<(abc a)const
    {
        return sum>a.sum;
    }
};
priority_queue<abc>heap;
int a[100010],b[100010];
int main()
{
    int n,i,j;
    cin>>n;
    for(i=0;i<n;i++)
        scanf("%d",a+i);
    for(i=0;i<n;i++)
        scanf("%d",b+i);
    sort(a,a+n);
    sort(b,b+n);
    for(i=0;i<n;i++)
        heap.push(abc(a[i]+b[0],0));
    for(i=0;i<n;i++)
    {
        abc it=heap.top();heap.pop();
        printf("%d%c",it.sum,i==n-1?'\n':' ');
        if(it.i+1<n)
            heap.push(abc(it.sum-b[it.i]+b[it.i+1],it.i+1));
    }
    return 0;
}