码迷,mamicode.com
首页 > 编程语言 > 详细

九度OJ—题目1534:数组中第K小的数字

时间:2014-12-19 19:14:30      阅读:160      评论:0      收藏:0      [点我收藏+]

标签:九度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;
}


九度OJ—题目1534:数组中第K小的数字

标签:九度oj   1534   数组中第k小的数字   

原文地址:http://blog.csdn.net/wtyvhreal/article/details/42029883

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!