蓝桥杯训练--力扣4

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

难度困难6295收藏分享切换为英文接收动态反馈

给定两个大小分别为 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

#思想

1. 通过extend函数把nums2添加到nums1里面

2. 用sort函数排序

3. 找到中间位置的下标

4. 偶数和奇数情况分俩种计算中位数

#代码

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        nums1.extend(nums2) 
        nums1.sort()
        mid = len(nums1) // 2
        if len(nums1) % 2 == 0: #偶数
            ans = (nums1[mid] + nums1[mid-1]) / 2
        else:
            ans = nums1[mid]
        return ans

#运行结果