python--寻找两个列表的中位数

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

# nums1 = [1,3,3]
# nums2 = [2,5,4]
def list_model(nums1,nums2):
    nums = nums1 + nums2
    nums.sort()
    print("您输入的两列表为 :",nums)
    n = len(nums)

    # print((nums[int(n/2-1)] + nums[int(n/2)])/2)
#     print(n)
    if n%2==0:
        model = (nums[int(n/2-1)] + nums[int(n/2)])/2
    else:
        model = nums[int((n+1)/2 - 1)]
    return(model)

    

x = input("请输入第一个列表 :")
nums1 = x.split(',')
nums1 = [float(nums1[i]) for i in range(len(nums1))]

y = input("请输入第二个列表 :")
nums2 = y.split(',')
nums2 = [float(nums2[i]) for i in range(len(nums2))]

print("您所求的两列表的中位数为 ：",list_model(nums1,nums2))

请输入第一个列表 :1,2.5,4
请输入第二个列表 :2,3,4
您输入的两列表为 : [1.0, 2.0, 2.5, 3.0, 4.0, 4.0]
您所求的两列表的中位数为 ： 2.75