给出已经排序数组,a和b,长度分别为n、m,请找出一个时间复杂度为O(log2(n+m))的算法,找到排在第k位置的元素。
#include <stdio.h>
int findk(int a[], int n, int b[], int m, int k);
int main(){
int i,result;
int a[]={1,5,8};
int b[]={2,3,4,6,7};
int n = sizeof(a)/sizeof(a[0]);
int m = sizeof (b)/sizeof(b[0]);
for(i=1; i <= n + m; i++){
result =findk(a,n,b,m,i);
printf("第%d小的元素是:%d\n",i,result);
}
}
//在两个升序排列的数组中找到第K大的元素
int findk(int a[], int n, int b[], int m, int k) {
if(k<=0)return -1;
if(n>m) //用于保证n<=m,即保证前一个数组的元素较少
return findk(b,m,a,n,k);
if(n==0)
return b[k-1];
if(k==1)
return a[0]<b[0]?a[0]:b[0];
int numa=(n>=k/2)?k/2:n; //当数组中没有k/2个元素时取n
int numb=k-numa;
if(a[numa-1]==b[numb-1])
return a[numa-1];
else if(a[numa-1]>b[numb-1])
return findk(a,n,&b[numb],m-numb,k-numb);
else
return findk(&a[numa],n-numa,b,m,k-numa);
}