九度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面试题

#include <stdio.h>
#include <stdlib.h>

long long a[100000];
long long b[100000];
long long n,m,i,j,k;

int compare(const void * p, const void * q){
    return *(long long *)p - *(long long *)q;
}  

long long cal(long long v){
    long long add=0,j=m-1;

    for(i=0;i<n;i++){
        while(j>=0&&(a[i]+b[j])>v)
			j--;
        add+=(j+1);
    }

    return add;
}

long long find(long long ll,long long rr,long long k){
    long long mid;
    while(ll<=rr){
        mid=(ll+rr)/2;
        if(k<=cal(mid))
			rr=mid-1;
        else
			ll=mid+1;
    }
    return ll;
}

int main()
{
    long long ll,rr;
    while(scanf("%lld%lld%lld",&n,&m,&k)!=EOF)
	{
        for(i=0;i<n;i++){
            scanf("%lld",&a[i]);
        }
		qsort(a, n, sizeof(long long), compare);  

        for(i=0;i<m;i++){
            scanf("%lld",&b[i]);
        }
		qsort(b, m, sizeof(long long), compare);

        ll=a[0]+b[0];
        rr=a[n-1]+b[m-1];

        printf("%lld\n",find(ll,rr,k));
    }

    return 0;
}