最小的N个和

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

思路就是各种排序。

首先把两组数存到a,b两个数组中,sort一下,使其有序。

而后,把a[1]和所有b的和加入到叫ans的大根堆中来,一共n个。

而后暴力枚举剩下的,恩,肯定超时,怎么办?如果a[i]与b[j]的和数不能加入到ans中,那就没有继续搜下面的j的必要了,因为a、b数组有序。

同样,如果某个i与j==1的和数也不能加入进ans,也没有继续搜下面的i的必要了。(这个并没有在代码中使用)

代码实现:

 1 #include<queue>
 2 #include<cstdio>
 3 #include<algorithm>
 4 using namespace std;
 5 int n,m,c,d,a[100010],b[100010];
 6 priority_queue <int> ans;
 7 int main(){
 8     scanf("%d",&n);
 9     for(int i=1;i<=n;i++) scanf("%d",&a[i]);
10     for(int i=1;i<=n;i++) scanf("%d",&b[i]);
11     sort(a+1,a+n+1);
12     sort(b+1,b+n+1);
13     for(int i=1;i<=n;i++) ans.push(a[1]+b[i]);
14     for(int i=2;i<=n;i++)
15     for(int j=1;j<=n;j++){
16         c=ans.top();
17         d=a[i]+b[j];
18         if(d<c) ans.pop(),ans.push(d);
19         else break;
20     }
21     for(int i=n;i>0;i--) a[i]=ans.top(),ans.pop();
22     for(int i=1;i<=n;i++) printf("%d ",a[i]);
23     printf("\n");
24     return 0;
25 }

挺简单的钻石题~

题目来源:CODE[VS]

posted @ 2017-01-01 20:13  J_william  阅读(298)  评论(0编辑  收藏  举报