Google面试题 数组中第K小的数字

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

时间限制：2 秒

内存限制：128 兆

特殊判题：否

提交：682

解决：105

题目描述：

给定两个整型数组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面试题

[cpp]  view plain copy

1.   

看这道题就要知道一定不能用一般的方法做，因为不能存放那么大的数组，这里用二分法，并且只要找的次数就可以与 k进行比较

[cpp]  view plain copy

1. #include<iostream>  
2. #include<stdio.h>  
3. #include<cstring>  
4. #include<math.h>  
5. #include<algorithm>  
6. using namespace std;  
7. int a[100005],b[100005];  
8. long long m,n,k;  
9. int judge(long long mi,long long k)  
10. {  
11.     long long i,j sum=0;  
12.     j=n-1;  
13.   
14.     for(i=0;i<m;i++)  
15.     {         
16.     while(a[i]+b[j]>mi&&j>=0)j--;  
17.   
18.       
19.     if(j<0)   break;  
20.     sum=sum+j+1;//这里利用每次比中值小的数组的个数来表示  
21.                 // 要比每次都加1好 如果从中间值找不到k个数   
22.     if(sum>=k)   return 1;//就把范围扩大1/4继续寻找，直到找到   
23.     }  
24.     return 0;  
25. }  
26. int main()  
27. {  
28.     int i;  
29.    long long R,L,mid;  
30.     while(scanf("%lld%lld%lld",&m,&n,&k)!=EOF)  
31.     {  
32.         memset(a,0,sizeof(a));  
33.         memset(b,0,sizeof(b));  
34.         for(i=0;i<m;i++)scanf("%d",&a[i]);  
35.         for(i=0;i<n;i++)scanf("%d",&b[i]);  
36.         sort(a,a+m);  
37.          sort(b,b+n);  
38.         L=a[0]+b[0];  
39.         R=a[m-1]+b[n-1];  
40.         while(L<R)  
41.         {  
42.             mid=(L+R+1)/2;  
43.             if(judge(mid-1,k)) R=mid-1;  
44.             else    L=mid;  
45.         }  
46.         printf("%lld\n",L);  
47.     }  
48.     return 0;  
49. }