美文网首页
04.leetcode题目讲解(Python):两个排序数组的中

04.leetcode题目讲解(Python):两个排序数组的中

作者: 夏山闻汐 | 来源:发表于2018-06-25 22:38 被阅读113次

    题目:

    给定两个大小为 m 和 n 的有序数组 nums1 和 nums2 。

    请找出这两个有序数组的中位数。要求算法的时间复杂度为 O(log (m+n)) 。

    示例 1:
    nums1 = [1, 3]
    nums2 = [2]
    中位数是 2.0
    示例 2:
    nums1 = [1, 2]
    nums2 = [3, 4]
    中位数是 (2 + 3)/2 = 2.5

    利用Python list 的 sort 函数, 这道题还是很容易实现,主要是需要判断list 元素个数是奇数还是偶数。Python3 编码参考如下:

    class Solution:
        def findMedianSortedArrays(self, nums1, nums2):
            """
            :type nums1: List[int]
            :type nums2: List[int]
            :rtype: float
            """
            self.nums = nums1 + nums2
            self.nums.sort()
            size = len(self.nums)
    
            if size % 2 == 1:
                return self.nums[size//2]
    
            if size % 2 == 0:
                return (self.nums[size//2] + self.nums[size//2 - 1])/2
    

    ps:如果您有好的建议,欢迎交流 :-D,也欢迎访问我的个人博客:tundrazone.com

    相关文章

      网友评论

          本文标题:04.leetcode题目讲解(Python):两个排序数组的中

          本文链接:https://www.haomeiwen.com/subject/uqmuyftx.html