Median of Two Sorted Array

There are two sorted arrays A and B 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)).

class Solution:
    # @return a float
    def findMedianSortedArrays(self, A, B):
        lenA = len(A); lenB = len(B)
        if (lenA + lenB) % 2 == 1:
            return self.getKth( A, B, (lenA + lenB) / 2 + 1)
        else :
            return (self.getKth( A, B, (lenA + lenB) / 2 ) + self.getKth( A, B, (lenA + lenB) / 2 + 1 ) ) * 0.5
    def getKth(self, A, B, k):
        lenA = len(A); lenB = len(B)
        if lenA > lenB : return self.getKth(B,A,k)
        if lenA == 0 : return B[k-1]
        if k == 1 : return min(A[0],B[0])
        pa = min(k/2, lenA) ; pb = k - pa
        if A[pa-1] <= B[pb-1] :
            return self.getKth(A[pa:], B, pb)
        else :
            return self.getKth(A,B[pb:], pa)
            
        


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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值