Codevs No.1245 最小的N个和

2016-05-31 18:52:15

题目链接: 最小的N个和 Codevs No.1245

题目大意:

　　给两个等长数列,各取一个数求和,找到最小的N组

解法:

　　堆优化的大暴力

　　直接枚举所有可能在最大堆中更新,删除最大组合

需要注意的地方:

　　1.如果任何一个加数大于等于堆顶元素,break

　　2.如果两者之和已经大于等于堆顶元素,break

//最小的N个和 (Codevs No.1245)
//堆(优先队列)
#include<stdio.h>
#include<algorithm>
#include<queue>
using namespace std;
const int maxn=100010;
priority_queue <int> q;
int N;
int x[maxn];
int y[maxn];
int ans[maxn];
int main()
{
    scanf("%d",&N);
    for(int i=1;i<=N;i++)
    {
        scanf("%d",&x[i]);
    }
    for(int j=1;j<=N;j++)
    {
        scanf("%d",&y[j]);
    }
    sort(x+1,x+N+1);
    sort(y+1,y+N+1);
    for(int i=1;i<=N;i++)q.push(x[i]+y[i]);
    for(int i=1;i<=N;i++)
    {
        if(x[i]>=q.top())break;
        for(int j=1;j<=N;j++)
        {
            if(i==j)continue;
            if(y[j]+x[i]>=q.top())break;
            q.push(x[i]+y[j]);
            q.pop();
        }
    }
    for(int i=1;i<=N;i++)
    {
        ans[i]=q.top();
        q.pop();
    }
    for(int i=N;i>=1;i--)
    {
        printf("%d ",ans[i]);
    }
    return 0;
}