codevs 1245 最小的N个和

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了codevs 1245 最小的N个和相关的知识,希望对你有一定的参考价值。

时间限制: 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;
}

 

 

以上是关于codevs 1245 最小的N个和的主要内容,如果未能解决你的问题,请参考以下文章

Codevs 1245 最小的N个和

codevs1245 最小的N个和

codevs 1245 最小的N个和

最小的N个和 Codevs No.1245

codevs 1245 最小的N个和

codevs 1245 最小的N个和