LeetCode:Median of two sorted array

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个。


  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值