leetcode记录4.Median of Two Sorted Arrays

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

hinanmu

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

阅读量81

点赞数

本文链接：https://blog.csdn.net/csdn_47/article/details/88142770

版权

leetcode 专栏收录该内容

37 篇文章 0 订阅

订阅专栏

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]