题目
题目描述
有两个长度都是 N 的序列 A 和 B,在 A 和 B 中各取一个数相加可以得到 N^2 个和,求这 N^2 个和中最小的 N 个。
输入输出格式
输入格式:
第一行一个正整数 N;
第二行 N 个整数 Ai, 满足 Ai<=Ai+1 且 Ai<=10^9;
第三行 N 个整数 Bi, 满足 Bi<=Bi+1 且 Bi<=10^9.
【数据规模】
对于 50% 的数据中,满足 1<=N<=1000;
对于 100% 的数据中,满足 1<=N<=100000。
输出格式:
输出仅一行,包含 N 个整数,从小到大输出这 N 个最小的和,相邻数字之间用空格隔开。
输入输出样例
输入样例 #1:
3
2 6 6
1 4 8
输出样例 #1:
3 6 7
题解
首先 a、b 从小到大排序。
众所周知,最小的数字就是 a[1]+b[1]
那么第二个数字一定是 a[1]+b[2]、a[2]+b[1]中的一个!
所以设当前最小的是 a[l]+b[r]
那么下一个答案就存在于 a[l]+b[r+1]和 a[l+1]+b[r]中。
所以我们搞个优先队列,每次取出最小的输出,并对它进行扩展。
复杂度 O(nlogn)
优先队列 STL 自带。
代码:
0 条评论