Leetcode #4 Median of Two Sorted Arrays

There are two sorted arrays nums1 and nums2 of size m and n respectively. Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).

借鉴了别人的经验,将问题转化为寻找第k小元素的问题。

     int findkth(int a[],int m,int b[],int n,int k)
    {
        if(m>n)
        return findkth(b,n,a,m,k);
        
        if(m == 0)
        return b[k-1];
        
        if(k==1)
        return a[0]<b[0]?a[0]:b[0];
        
        
        int pa = m<k/2?m:k/2;
        int pb = k - pa;
        
        if(a[pa-1] > b[pb-1])
           return findkth(a,m,b+pb,n-pb,k-pb);
        else if(a[pa-1] < b[pb-1])
           return findkth(a+pa,m-pa,b,n,k-pa);
           else return a[pa-1];
    }
    
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) {
    
    int total = nums1Size + nums2Size;
    
    if(total%2 == 1)
    return findkth(nums1,nums1Size,nums2,nums2Size,total/2+1);
    
    else
     return (findkth(nums1,nums1Size,nums2,nums2Size,total/2) + findkth(nums1,nums1Size,nums2,nums2Size,total/2+1))*1.0/2;
    
    


}

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值