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

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

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

 

nums2 = [1,]
nums1 = [54,78,90,]

def test(nums):
    lengh = len(nums)
    if lengh % 2 == 0:
        index = [lengh / 2 - 1, lengh / 2]
    else:
        index = [lengh // 2]
    if len(index) == 2:
        # print(index)
        print(nums[int(index[0])], nums[int(index[1])])
    else:
        print(nums[int(index[0])])


if not nums2 or not nums1:
    if nums1:
        test(nums1)
    else:
        test(nums2)
else:
    lengh = len(nums1) + len(nums2)
    index = []
    if lengh == 1:
        index.append(0)
    elif lengh % 2 != 0:
        index.append((lengh - 1) // 2)
    else:
        index.append(lengh // 2 - 1)
        index.append(lengh // 2)

    result = []
    for it in range(index[-1] + 1):
        if nums1 and nums2:
            if nums1[0] > nums2[0]:
                result.append(nums2.pop(0))
        else:
            result.append(nums1.pop(0))

    print(result)
    if len(index) == 1:
        print(result[-1])
    else:
        print(result[-1] / 2 + result[-2] / 2)