4.LeetCode-寻找两个有序数组的中位数-Python

weixin_39883139

于 2020-02-13 18:10:53 发布

阅读量120

点赞数

分类专栏： leetcode 文章标签： leetcode python

本文链接：https://blog.youkuaiyun.com/weixin_39883139/article/details/104299757

版权

leetcode 专栏收录该内容

14 篇文章

订阅专栏

本文探讨了在LeetCode上解决寻找两个有序数组中位数的问题，提供了Python代码实现。通过对数组进行排序并根据长度的奇偶性计算中位数，确保了O(log(m+n))的时间复杂度。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

4.LeetCode-寻找两个有序数组的中位数-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

解法：当长度为偶数和奇数时，分别判断。只要注意下nums[x]中x不能是float，类型转换下即可。
难点：用到Python的sorted函数。

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        nums = sorted(nums1+nums2)
        if len(nums)%2 ==0:
            answer = (nums[int(len(nums)/2-1)]+nums[int(len(nums)/2)])/2
        else:
            answer = nums[int((len(nums)+1)/2)-1]
        return answer