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

题目:

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

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。