Python实现两个排序数组的中位数

最新推荐文章于 2024-07-15 16:28:32 发布

少儿西笑

最新推荐文章于 2024-07-15 16:28:32 发布

阅读量644

点赞数

分类专栏：算法文章标签：算法

本文链接：https://blog.youkuaiyun.com/weixin_42936560/article/details/87694298

版权

这篇博客探讨了如何在Python中找到两个已排序数组合并后的中位数。通过递归方法，当数组元素总数为奇数时，确定中位数的下标为n/2，偶数时则为(n/2-1)和n/2位置元素的平均值。内容引用了优快云上的一篇文章作为参考。

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

Q: 目前有两个已经排序好的数组(从⼩小到⼤大) A,B, 请求出两个数组合并之后的median。

思路：当两个数组合并后的总元素长度是奇数时，中位数的下标是n/2。当两个数组合并后的总元素个数是偶数时，中位数是下标n/2-1和下标n/2两个元素的平均值。运用递归法求A,B中第k个值。

class Solution:
    """
    @param A: An integer array.
    @param B: An integer array.
    @return: a double whose format is *.5 or *.0
    """
    def findMedianSortedArrays(self, A, B):
        n = len(A) + len(B)
        if n % 2 == 1:
            return self.findKth(A, B, n / 2 + 1)
        else:
            smaller = self.findKth(A, B, n / 2)
            bigger = self.findKth(A, B, n / 2 + 1)
            return (smaller + bigger) / 2.0

    def findKth(self, A, B, k):
        if len(A) == 0:
            return B[k - 1]
        if len(B) == 0:
            return A[k - 1]
        if k == 1:
            return min(A[0], B[0])
        
        a = A[k / 2 - 1] if len(A) >= k / 2 else None