-
题目描述:
-
给定两个整型数组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;
}