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)).

Solution: 求两个有序数组的中位数,采用分治法,每次缩小搜索空间,分治法的基准点选择两个数组中位数。 该问题要注意m+n分别为奇偶的情况。

class Solution {
public:
    int findk(vector<int>& a, vector<int>& b, int k){
        int n = a.size();
        int m = b.size();
        if (n <= 0) return b[k-1];  
        if (m <= 0) return a[k-1];  
        if (k <= 1) return min(a[0], b[0]);   
        if (b[m/2] >= a[n/2])  
        {  
            if ((n/2 + 1 + m/2) >= k) { 
                  vector<int> b2(b.begin(), b.begin() + m/2);
                  return findk(a, b2, k); 
            }
            else{
                 vector<int> a2(a.begin() + n/2 + 1, a.end());
                 return findk(a2, b, k - (n/2 + 1));  
            }  
                 
        }  
        else  
        {  
            if ((m/2 + 1 + n/2) >= k){
                 vector<int> a2(a.begin(), a.begin() + n/2);
                 return findk( a2,b, k);
            }
            else{  
                 vector<int> b2(b.begin() + m/2 + 1, b.end());
                 return findk( a,b2,k - (m/2 + 1));  
            }
        }  
    }
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int m = nums1.size();
        int n = nums2.size();
        if((m + n ) %2 == 0)
        {
            return (findk(nums1, nums2,(m + n)/2 ) + findk(nums1, nums2,(m + n)/2 + 1 )) / 2.0;
        }
        else
            return findk(nums1, nums2,(m + n)/2 + 1);
    }
};


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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值