九度oj题目1534：数组中第K小的数字

题目描述：

给定两个整型数组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
****************************************************************/