Leetcode 4. 寻找两个正序数组的中位数（困难）

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

给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。

示例 1：
输入：nums1 = [1,3], nums2 = [2]
输出：2.00000
解释：合并数组 = [1,2,3] ，中位数 2

示例 2：
输入：nums1 = [1,2], nums2 = [3,4]
输出：2.50000
解释：合并数组 = [1,2,3,4] ，中位数 (2 + 3) / 2 = 2.5

示例 3：
输入：nums1 = [0,0], nums2 = [0,0]
输出：0.00000

示例 4：
输入：nums1 = [], nums2 = [1]
输出：1.00000

示例 5：
输入：nums1 = [2], nums2 = []
输出：2.00000

提示：

• nums1.length == m
• nums2.length == n
• 0 <= m <= 1000
• 0 <= n <= 1000
• 1 <= m + n <= 2000
• -10^6 <= nums1[i], nums2[i] <= 10^6

解题方法，先合并（由于数组已排好序，从后往前合并即可）再求中值

• Python

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        merge = []
        if not nums1:
            if len(nums2) % 2 == 0:
                return (nums2[(len(nums2) // 2)] + nums2[(len(nums2) // 2) - 1]) / 2
            else:
                return nums2[(len(nums2) // 2)]
        elif not nums2:
            if len(nums1) % 2 == 0:
                return (nums1[(len(nums1) // 2)] + nums1[(len(nums1) // 2) - 1]) / 2
            else:
                return nums1[(len(nums1) // 2)]
        else:
            m,n = len(nums1), len(nums2)
            for i in range(m+n):
                merge.append(0)
            #print(len(merge))
            while m>0 and n>0:
                if nums1[m-1] > nums2[n-1]:
                    merge[m+n-1] = nums1[m-1]
                    m = m - 1
                else:
                    merge[m+n-1] = nums2[n-1]
                    n = n - 1
            if m > 0:
                merge[:m] = nums1[:m]
            elif n > 0:
                merge[:n] = nums2[:n]
            if len(merge) % 2 == 0:
                return (merge[(len(merge) // 2)] + merge[(len(merge) // 2) - 1]) / 2
            else:
                return merge[(len(merge) // 2)]