double findkth(int *nums1,int nums1Size,int *nums2,int nums2Size,int k){
int a,b;
if(nums1Size<nums2Size){
a=(k/2<nums1Size)?k/2:nums1Size;
b=k-a;
}
else{
b=(k/2<nums2Size)?k/2:nums2Size;
a=k-b;
}
if(nums1Size==0)return nums2[k-1];
if(nums2Size==0)return nums1[k-1];
if(k==1)return (nums2[k-1]>nums1[k-1])?nums1[k-1]:nums2[k-1];
if(nums1[a-1]<nums2[b-1])return findkth(nums1+a,nums1Size-a,nums2,nums2Size,k-a);
else if(nums1[a-1]==nums2[b-1])return nums1[a-1];
else return findkth(nums1,nums1Size,nums2+b,nums2Size-b,k-b);
}
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) {
int k=nums1Size+nums2Size;
if(k%2==0)//printf("%f %f",findkth(nums1,nums1Size,nums2,nums2Size,k/2), findkth(nums1,nums1Size,nums2,nums2Size,k/2+1));
return (findkth(nums1,nums1Size,nums2,nums2Size,k/2)+findkth(nums1,nums1Size,nums2,nums2Size,k/2+1))/2;
else //printf("%f",findkth(nums1,nums1Size,nums2,nums2Size,k/2));
return findkth(nums1,nums1Size,nums2,nums2Size,k/2+1);
}
求两个排序数组的中位数:这是一个O(log(n+m))的算法。将求中位数转化为求两个数组中第(m+n)/2大的数,进一步扩展为求第k大的数。
首先考虑三种边界情况:两个数组中有一个数组长度为0,则返回另一个数组的中位数即可;若k=1,则只需输出两个数组各自最小值中较小的那一个。
然后,要求第k大的数,先假设两个数组各取前k/2,如果,第一个数组中第k/2个数字小于第二个数组中第k/2个数字,则说明第一个数组中前k/2个都小于整体第k个,故只需在后半部分找。反之同理。若 刚好相等,则该值为第k个。