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

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

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

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

一、代码

def midnum(nums1,nums2):
    nums1.extend(nums2)#合并
    nums1.sort()#排序
    n=len(nums1)
    index=n//2
    if n%2==1:
        print(nums1[index])
    else:
        print((nums1[index-1]+nums1[index])/2)




if __name__=='__main__':
    nums1 = input().split()
    nums2 = input().split()
    nums1 = [int(x) for x in nums1 if x.isdigit()]
    nums2 = [int(x) for x in nums2 if x.isdigit()]
    midnum(nums1,nums2)

二、结果