Leetcode做题记录-4. 寻找两个正序数组的中位数

题目描述：
给定两个大小为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的中位数。
进阶：你能设计一个时间复杂度为 O(log (m+n)) 的算法解决此问题吗？
来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/median-of-two-sorted-arrays

根据进阶要求的时间复杂度为O(log(m+n)),考虑用二分法，思路见代码注释。

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        #递归加上二分法查找
        def findKthElement(arr1,arr2,k):
            len1,len2 = len(arr1),len(arr2)
            #边界条件
            if len1 > len2:
                return findKthElement(arr2,arr1,k)#使得arr1数组的长度小于arr2的数组的长度
            #边界条件
            if not arr1:
                return arr2[k-1]#如果列表arr1为空，则直接返回中值
            #边界条件
            if k == 1:
                return min(arr1[0],arr2[0])#当k为1的时候，表示两个列表中第一元素小的那个就是需要的k最小值

            i,j = min(k//2,len1)-1,min(k//2,len2)-1#根据k，分别对两个列表进行划分，并判断当前的列表的长度与k//2之间的关系.由于索引从0开始，所以需要-1
            #对不要要的部分进行舍弃（本部分一定处在k最小值得左边）,给定的两个数组是有序的
            if arr1[i] > arr2[j]:
                return findKthElement(arr1,arr2[j+1:],k-j-1)#例如，arr1的0-i的元素值为34567，arr2的0-j的元素值为12345，那么，他的中位数一定不会在arr2的0-j上
            else:
                return findKthElement(arr1[i+1:],arr2,k-i-1)
                
        l1,l2 = len(nums1),len(nums2)
        left,right = (l1+l2+1)//2,(l1+l2+2)//2#总长度为奇数的时候，left和right的值会相等，若为偶数，则两个值不会相等
        return (findKthElement(nums1,nums2,left)+findKthElement(nums1,nums2,right))/2