LeetCode : No4 Median of Two Sorted Arrays

来源：互联网发布：三洋air9s性能数据编辑：程序博客网时间：2024/04/30 08:29

题目链接：

https://leetcode.com/problems/median-of-two-sorted-arrays/

该问题可以这样描述：在两个升序排列数组中，寻找第k小的数。中位数就是(m+n)/2小的数。则可以通过比较A[k/2-1]和B[k/2-1]的大小，剔除前k/2个较小的数字。在寻找第k-k/2小的数字，通过递归的方式，直至找到第(m+n)/2个值。

耗时：140ms

算法复杂度：O(log(m+n))

class Solution:    # @return a float    def findMedianSortedArrays(self, A, B):        m = len(A)        n = len(B)        total = m + n        if total%2 == 1:            return float(self.findK(A,m,B,n,total/2+1))        else:            return float(self.findK(A,m,B,n,total/2)+self.findK(A,m,B,n,total/2+1))/2    def findK(self, A, m, B, n, k):        if (m>n):            return self.findK(B,n,A,m,k)        elif (m==0):            return B[k-1]        if k==1:            return min(A[0],B[0])        pa = min(k/2,m)        pb = k - pa        if (A[pa-1] < B[pb-1]):            return self.findK(A[pa:],m-pa,B,n,k-pa)        elif (A[pa-1] > B[pb-1]):            return self.findK(A,m,B[pb:],n-pb,k-pb)        else:            return A[pa-1]

0 0