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

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

原始题目链接：https://leetcode-cn.com/problems/median-of-two-sorted-arrays/

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

算法的时间复杂度应该为 O(log (m+n)) 。

示例 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

解题思路：

这道题要求O(log(m+n))，这个难度是挺难的，有序会想到使用二分查找，转换成了寻找第k小的数，分别是在两个数组中查找，具体实现看代码及注释和参考文献的图解。

代码实现：

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        def helper(k):
            index1, index2 = 0, 0
            while True:
                # 边界情况，第一个数组的长度为0或者index1的索引到头了即等于m
                # 满足条件的第k个值在nums2中的第index2 + k - 1个元素
                if index1 == m:
                    return nums2[index2 + k - 1]
                # 边界情况，第二个数组的长度为0或者index2的索引到头了即等于n
                if index2 == n:
                    # 满足条件的第k个值在nums1中的第index1 + k - 1个元素
                    return nums1[index1 + k - 1]
                # 找到第k个小的元素了
                if k == 1:
                    return min(nums1[index1], nums2[index2])
                
                # 根据k来计算开始比较的第几个索引
                new_index1 = min(index1 + k // 2 - 1, m - 1)
                new_index2 = min(index2 + k // 2 - 1, n - 1)
                pivot1, pivot2 = nums1[new_index1], nums2[new_index2]
                if pivot1 <= pivot2:
                    # 更新index1和k的值，删除比第 k 小的元素要小的元素，一共删除了new_index1 - index1 + 1
                    # 因为删除了元素，所以原来是第k个，但现在需要减去删除元素的个数：new_index1 - index1 + 1
                    k -= new_index1 - index1 + 1
                    # 从删除之后的元素开始后面的另一次新的计算
                    index1 = new_index1 + 1
                else:
                    k -= new_index2 - index2 + 1
                    index2 = new_index2 + 1
        
        m = len(nums1)
        n = len(nums2)
        total = m + n

        # 偶数个元素
        if total % 2 == 0:
           return (helper(total // 2 + 1) + helper(total // 2)) / 2
        else:
            return helper(total // 2 + 1) 

参考文献：
https://leetcode-cn.com/problems/median-of-two-sorted-arrays/solution/xun-zhao-liang-ge-you-xu-shu-zu-de-zhong-wei-s-114/