2.寻找两个正序数组的中位数_Python算法

给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。算法的时间复杂度应该为 O(log (m+n)) 。

示例 1：

输入：nums1 = [1,3], nums2 = [2]
输出：2.00000
解释：合并数组 = [1,2,3] ，中位数 2

示例 2：

输入：nums1 = [1,2], nums2 = [3,4]
输出：2.50000
解释：合并数组 = [1,2,3,4] ，中位数 (2 + 3) / 2 = 2.5

题解：

# 定义解题方法
def fun1(nums1, nums2):
    nums1 = list(nums1)
    nums2 = list(nums2)
    nums3 = nums1 + nums2  # 根据题目要求进行先合并
    data = 0.0  # 题目要求返回一个小数
    n3_len = len(nums3)
    nums3.sort()
    if n3_len % 2 == 0: #整除表示个数为偶数
        mid = n3_len // 2 #其中一个中间数下标
        data = (nums3[mid - 1] + nums3[mid]) / 2 #根据题目要求中间2个数相加后平均
    else:
        mid = (n3_len - 1) // 2
        data = nums3[mid] #根据题目要求取中间数
    return float(data)

# 根据题目输入例子进行验证
nums1 = [1, 3]
nums2 = [2]
print(fun1(nums1, nums2))