leetcode 日经贴，python code -Median of Two Sorted Arrays

bachelorchen

于 2015-02-17 13:49:38 发布

阅读量342

点赞数

CC 4.0 BY-SA版权

分类专栏： python 算法 leetcode

本文链接：https://blog.youkuaiyun.com/bachelorchen/article/details/43866155

算法同时被 3 个专栏收录

183 篇文章

订阅专栏

leetcode

173 篇文章

订阅专栏

python

47 篇文章

订阅专栏

本文介绍了一种高效算法，用于查找两个已排序数组中的中位数。通过递归地划分数组并比较中间元素，逐步缩小搜索范围，最终找到目标中位数。此算法适用于计算机科学中的数据处理和算法设计等领域。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Median of Two Sorted Arrays

class Solution:
    # @return an element from A or B
    def findKth(self, A, B, k):
        n, m = len(A), len(B)
        if n == 0: return B[k - 1]
        if m == 0: return A[k - 1]
        mid1, mid2 = n / 2, m / 2
        if mid1 + mid2 + 2 > k:
            if A[mid1] >= B[mid2]:
                return self.findKth(A[:mid1], B, k)
            else:
                return self.findKth(A, B[:mid2], k)
        else:
            if A[mid1] >= B[mid2]:
                return self.findKth(A, B[mid2 + 1:], k - mid2 - 1)
            else:
                return self.findKth(A[mid1 + 1:], B, k - mid1 - 1)
                
    # @return a float
    def findMedianSortedArrays(self, A, B):
        n = len(A) + len(B)
        if n % 2 == 0:
            return (self.findKth(A, B, n / 2) + self.findKth(A, B, n / 2 + 1)) / 2.0
        else:
            return self.findKth(A, B, n / 2 + 1)