T1245 最小的N个和 codevs

 空间限制: 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。

 
#include <algorithm>
#include <iostream>
#include <cstdio>
#include <queue>
#define maxn 100015

using namespace std;

int n,a[maxn],b[maxn],tot;
priority_queue<int,vector<int>,less<int> >que;

int main()
{
    scanf("%d",&n);
    for(int i=1;i<=n;i++)
        scanf("%d",&a[i]);
    for(int i=1;i<=n;i++)
        scanf("%d",&b[i]);
    sort(a+1,a+n+1);
    sort(b+1,b+n+1);
    for(int i=1;i<=n;i++)
        for(int j=1;j<=n;j++)
        {
            int cnt=a[i]+b[j];
            if(tot<=n)
            {
                que.push(cnt);
                tot++;
            }
                
            else
            {    
                if( cnt>=que.top() )
                {
                    //tot--;
                    break;
                }
                else
                {
                    //tot--;
                    que.pop();
                    que.push(cnt);
                }    
            }
                
        }
    int ans[maxn];
    for(int i=1;i<=tot;i++)
    {
        ans[i]=que.top();
        que.pop();
    }
    sort(ans+1,ans+tot+1);
    for(int i=1;i<tot;i++)
        cout<<ans[i]<<" ";
    return 0;
}

 

posted @ 2017-03-04 20:47  Aptal丶  阅读(170)  评论(0编辑  收藏  举报