leetcode 4. 寻找两个有序数组的中位数

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。

请你找出这两个有序数组的中位数，并且要求算法的时间复杂度为 O(log(m + n))。

你可以假设 nums1 和 nums2 不会同时为空。

示例 1:

nums1 = [1, 3]
nums2 = [2]

则中位数是 2.0
示例 2:

nums1 = [1, 2]
nums2 = [3, 4]

则中位数是 (2 + 3)/2 = 2.5

 

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        if len(nums1) <= len(nums2):
            m, n = len(nums1), len(nums2)
        else:
            m, n = len(nums2), len(nums1)
            nums1, nums2 = nums2, nums1
        left, right = 0, m
        half = (m+n+1) // 2
        while left <= right:
            i = (left + right) // 2
            j = half - i
            if i >= 1 and nums1[i-1] > nums2[j]:
                right = i - 1
            elif i < m and nums1[i] < nums2[j-1]:
                left = i + 1
            else:
                if i == 0:
                    leftmax = nums2[j-1]
                elif j == 0:
                    leftmax = nums1[i-1]
                else:
                    leftmax = max(nums2[j-1], nums1[i-1])
                if (m+n) % 2 == 1:
                    return leftmax
                if i == m:
                    rightmin = nums2[j]
                elif j == n:
                    rightmin = nums1[i]
                else:
                    rightmin = min(nums2[j], nums1[i])
                return (leftmax+rightmin) / 2