codevs 1245 最小的N个和

Posted 2020-09-02

时间限制: 1 s

 空间限制: 128000 KB

 题目等级 : 钻石 Diamond

题目描述 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。

此题做法等同于洛谷1631序列合并

大根堆  

最后用栈存一下输出（如果大佬有其他方法翻转过来，欢迎评论，我将吸取广泛意见）。

屠龙宝刀点击就送

#include <algorithm> 
#include <cstdio>
#include <stack>

using namespace std;

stack<int>stac;
int size,n,i,j,a[100000],heap[100000],b[100000];
inline void push(int a)
{
    heap[++size]=a;
    int pos=size;
    while(pos>1)
    {
        int next=pos/2;
        if(heap[next]>heap[pos]) break;
        swap(heap[next],heap[pos]);
        pos=next;
    }
}
inline void down()
{
    int pos=1;
    while(pos*2<=size)
    {
        int next=pos*2;
        if(heap[next+1]>heap[next])
        next++;
        if(heap[next]<heap[pos]) break;
        swap(heap[pos],heap[next]);
        pos=next;
    }
}
inline void pop()
{
    heap[1]=heap[size--];
    int pos=1;
    while(pos*2<=size)
    {
        int next=pos*2;
        if(heap[next+1]>heap[next]) next++;
        if(heap[next]<heap[pos]) break;
        swap(heap[next],heap[pos]);
        pos=next;
    }
}
int main()
{
    scanf("%d",&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)
    {
        for(j=0;j<n;++j)
        {
            if(size<n)
            push(a[i]+b[j]);
            else
            {
                int Another=a[i]+b[j];
                if(Another<heap[1])
                {
                    heap[1]=Another;
                    down();
                }
                else break;//剪枝
            }
        }
    }
    for(i=0;i<n;++i)
    {
        stac.push(heap[1]);
        pop();
    }
    while(!stac.empty())
    {
        printf("%d ",stac.top());
        stac.pop();
    }
    return 0;
}