第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,1e8]。
输出格式对应每个测试案例，
输出由A和B中元素两两相加得到的数组c中第K小的数字。
样例输入

2 2 3
1 2
3 4
3 3 4
1 2 7
3 4 5
样例输出

5
6

#include <stdio.h>

#include <iostream>

#include <algorithm>

#include <math.h>

#include <string.h>

using namespace std;

int a[];

int b[];

int n,m,k;

int cal(int v)

{

    int l,r,mid,i,d=;

    int min,max;

    for(i=;i<=n;i++)

    {

        min=a[i]+b[];

        max=a[i]+b[m];

        if(v<min)

            break;

        if(v>=max)

        {

            d+=m;

            continue;

        }

        l=,r=m;

        while(l<=r)

        {

            mid=(l+r)/;

            if(v<(a[i]+b[mid])) r=mid-;

            else                 l=mid+;

        }

        if(v!=(a[i]+b[l]))    l--;

        d += l;

    }

    return d;

}

int find(int l,int r,int k)

{

    int mid,i;

    while(l<=r)

    {

        mid=(l+r)/;

        if(k<=cal(mid)) r=mid-;

        else             l=mid+;

    }

    return l;

}

int main()

{

    int i,j,l,r;

    while(scanf("%d %d %d",&n,&m,&k)!=EOF)

    {

    for(i=;i<=n;i++)

        scanf("%d",&a[i]);

    for(i=;i<=m;i++)

        scanf("%d",&b[i]);

    sort(&a[],&a[n+]);

    sort(&b[],&b[m+]);

    l=a[]+b[];

    r=a[n]+b[m];

    printf("%d\n",find(l,r,k));

    }

    return ;

}