解题思路-leetcode第四题：寻找两个有序数组的中位数

解题思路-leetcode第四题：寻找两个有序数组的中位数

题目描述：给定两个大小为 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

看到题目首先想到的是把两个数组合并排序，再寻找中位数，但是一想时间复杂度为O(n)超过了O(log(m + n)，所以选择了使用三个指针来解题，设三个指针i,j,k。i用来指向数组一的遍历位置，j用来指向数组二的遍历位置，k用来判断总数组遍历位置是否超过中位数的位置。程序主题为一个while循环，循环条件为遍历位置没有超过中位数位置。然后为一个判断语句，如果遍历没有超过数组二的长度，或者没有超过数组一的长度且数组一的值小于数组二的值，则数组一向后移动一位，否则数组二向后移动一维，而不管哪种情况总数组都向后移动一位。程序中设置两个变量r1,r2，用于当长度为偶数时，存储中位数前后两个数的值。最后，当长度为奇数时，程序返回r2，当长度为偶数时，程序返回r1+r2的平均值。代码如下：

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        m, n = len(nums1), len(nums2)
        i, j, k = 0, 0, 0
        r1, r2 = 0, 0
        while k <= (m+n)//2:
            if j>=n or (i<m and nums1[i] < nums2[j]):
                r1, r2 = r2, nums1[i]
                i += 1
                k += 1      
            else:
                r1, r2 = r2, nums2[j]
                j += 1
                k += 1
        return float(r2) if (m+n)%2 != 0 else (r1 + r2)/2

提交后，代码通过。