给定两个整型数组A和B。我们将A和B中的元素两两相加可以得到数组C。 譬如A为[1,2]，B为[3,4].那么由A和B中的元素两两相加得到的数组C为[4,5,5,6]。 现在给你数组A和B，求由A和B两两相加得到的数组C中，第K小的数字。
输入：
输入可能包含多个测试案例。 对于每个测试案例，输入的第一行为三个整数m，n， k(1<=m,n<=100000， 1<= k <= n *m)：n，m代表将要输入数组A和B的长度。 紧接着两行， 分别有m和n个数， 代表数组A和B中的元素。数组元素范围为[0,1e9]。
输出：
对应每个测试案例， 输出由A和B中元素两两相加得到的数组c中第K小的数字。
样例输入： 2 2 3 1 2 3 4 3 3 4 1 2 7 3 4 5 样例输出： 5 6 来源： Google面试题
本来是用类似归并排序的原理，将A数组中每个数都分别加到B组上，得到一个N组已经从小到大排列好的数组，每次扫描这N个数组的最小值，直到得到第K个数，其复杂度为N*K，毫无疑问的超时了。。。。。。
后来借鉴了别人的代码，采用二分法由当前数组C最小值到最大值，统计小于当前mid的数字有多少个，慢慢逼近第K个数的值。
我们并不知道当前mid是否是数组C中的元素，但是，如果假设C[k]<mid,那么找得到mid2，满足c[k]<=mid2<mid，使mid2也满足数组C中小于mid2的元素的个数满足题目需要，二分法可以不断逼近C[k]，直到等于C[k]
#include<stdio.h> #include<algorithm> using namespace std; long long m, n, k; long long vm[100001], vn[100001]; long long count(long long mid) { long long ct = 0; long long kj = n - 1; for (int i = 0; i < m; i++) { while (kj >=0 && vm[i] + vn[kj] > mid) kj--; ct += (kj + 1); } return ct; } int main() { while (scanf("%lld%lld%lld", &m, &n, &k) != EOF) { for (int i = 0; i < m; i++) scanf("%lld", vm + i); for (int i = 0; i < n; i++) scanf("%lld", vn + i); sort(vm, vm + m); sort(vn, vn + n); long long be = vn[0] + vm[0]; long long en = vn[n-1] + vm[m-1]; long long mid,midt; while (be<=en) { mid = ((en-be)>>1)+be; if (k <= count(mid)) { midt = mid; en = mid-1; } else be = mid+1; } printf("%lld\n", midt); } return 0; } /************************************************************** Problem: 1534 User: 午夜小白龙 Language: C++ Result: Accepted Time:820 ms Memory:2584 kb ****************************************************************/
