leetcode记录4.Median of Two Sorted Arrays

最新推荐文章于 2023-01-26 14:28:30 发布

原创最新推荐文章于 2023-01-26 14:28:30 发布 · 107 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

37 篇文章

订阅专栏

本文介绍了一个算法，用于查找两个已排序数组的中位数。通过定义一个findMedianSortedArrays方法，该方法接收两个整数列表作为输入，并返回它们合并后的中位数。在实现中，我们首先检查总元素数量的奇偶性来确定中位数是单个值还是两个值的平均。然后使用findk辅助方法找到第k小的元素。

Median of Two Sorted Arrays

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        if (len(nums1) + len(nums2)) % 2 == 1:
            return self.findk(nums1, nums2, (len(nums1) + len(nums2) + 1) // 2)
        else:
            return (self.findk(nums1, nums2, (len(nums1) + len(nums2)) // 2) + self.findk(nums1, nums2, (len(nums1) + len(nums2)) // 2 + 1)) / 2
        
        
    def findk(self, nums1, nums2, k):
        l1= len(nums1)
        l2 = len(nums2)

        if l1 > l2:
            return self.findk(nums2, nums1, k)
        if l1 == 0:
            return nums2[k-1]
        if k == 1:
            return min(nums1[0], nums2[0])

        a = min(k // 2, l1)
        b = k - a
        if nums1[a - 1] < nums2[b- 1]:
            return self.findk(nums1[a:], nums2, k - a)
        elif nums1[a - 1] > nums2[b- 1]:
            return self.findk(nums1, nums2[b:], k - b)
        else:
            return nums1[a - 1]