LeetCode(4)求两个数列的中位数

LeetCode(4)求两个数列的中位数

问题

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

进阶：你能设计一个时间复杂度为 O(log (m+n)) 的算法解决此问题吗？

提示：

nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-106 <= nums1[i], nums2[i] <= 106

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/median-of-two-sorted-arrays
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

示例

示例 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

示例 3：

输入：nums1 = [0,0], nums2 = [0,0]
输出：0.00000

示例 4：

输入：nums1 = [], nums2 = [1]
输出：1.00000

示例 5：

输入：nums1 = [2], nums2 = []
输出：2.00000

题解代码

# 作者:爱编程的章老师
# 主题:LeetCode题解
# day04 : 找两个数列中所有元素的中位数

# 主功能函数
def find_median(nums1:list, nums2:list):
    m = len(nums1)
    n = len(nums2)
    left = (m+n+1)//2
    right = (m+n+2)//2
    return (find_kth(nums1, 0, nums2, 0, left) + find_kth(nums1, 0, nums2, 0, right))/2

# 找第k位数
def find_kth(nums1, i, nums2,j, k):
    # 如果nums1是空的,直接返回中nums2的中位数
    if i >= len(nums1):
        return nums2[j+k-1]
    # 如果nums2是空的,直接返回中nums2的中位数
    if j >= len(nums2):
        return nums1[i+k-1]
    
    # 如果k == 1 ,就返回当前要比较的值
    if k == 1:
        return min(nums1[i], nums2[j])
    
    ## 如果索引超过范围,设定为最大数
    if i +k//2 -1<len(nums1):
        midval1 = nums1[i+k//2-1]
    else:
        midval1 = 10**6 + 1
    
    # 如果索引超过范围,设定为最大数
    
    # 检测有没有还有空间的列表
    if j +k//2 -1<len(nums2):
        midval2 = nums2[j+k//2-1]
    else:
        midval2 = 10**6 + 1 
        
    # 小的那一边所以数列直接去掉,余下的购成新的数列,重复上面的逻辑即可
    if midval1<midval2:
        return find_kth(nums1, i+k//2, nums2, j, k-k//2)
    else:
        return find_kth(nums1, i, nums2, j+k//2, k-k//2)

后记

考试到题中要求时间 复杂度为log(m+n)
因此就不好用穷举法来找了,应该用二分法来查找.