4. 寻找两个正序数组的中位数（Python）

给定两个大小为 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:
		nums = nums1 + nums2
        nums.sort()
        n = len(nums)
        if n % 2 != 0:
            return float(nums[n // 2])
        return (nums[n // 2] + nums[n // 2 - 1]) / 2

方法二：双指针，通过双指针将两个数组合并成一个新数组，再求中位数。

class Solution:
    def findMedianSortedArrays(self, nums1: 'List[int]', nums2: 'List[int]') -> float:
        m = len(nums1)
        n = len(nums2)
        i, j = 0, 0
        result = []
        while i < m or j < n:
            if len(result) > (m + n) // 2:
                break
            if i < m and j < n:
                if nums1[i] > nums2[j]:
                    result.append(nums2[j])
                    j += 1
                else:
                    result.append(nums1[i])
                    i += 1
            elif i >= m:
                result.append(nums2[j])
                j += 1
            elif j >= n:
                result.append(nums1[i])
                i += 1
        if (m + n) % 2 == 1:
            return result[(m + n) // 2]
        else:
            return (result[(m + n) // 2] + result[(m + n) // 2 - 1]) / 2