leetcode 4. 寻找两个正序数组的中位数(python)52ms

最新推荐文章于 2022-07-08 08:19:09 发布

lanmengyiyu

最新推荐文章于 2022-07-08 08:19:09 发布

阅读量345

点赞数

分类专栏： leetcode 文章标签：算法 leetcode python 数组

本文链接：https://blog.csdn.net/lanmengyiyu/article/details/106682613

版权

leetcode 专栏收录该内容

30 篇文章 0 订阅

订阅专栏

题目描述：

给定两个大小为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。
请你找出这两个正序数组的中位数，并且要求算法的时间复杂度为 O(log(m + n))。
你可以假设 nums1 和 nums2 不会同时为空。

示例：

nums1 = [1, 3]
nums2 = [2]
则中位数是 2.0

nums1 = [1, 2]
nums2 = [3, 4]
则中位数是 (2 + 3)/2 = 2.5

解题思路：

class Solution(object):
    def findMedianSortedArrays(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: float
        """
        idx, jdx = 0, 0
        nums3 = []
        while (idx < len(nums1) and jdx < len(nums2)):
            if nums1[idx] <= nums2[jdx]:
                nums3.append(nums1[idx])
                idx = idx + 1 
            else:
                nums3.append(nums2[jdx])
                jdx = jdx + 1
        if idx == len(nums1) and jdx < len(nums2):
            while(jdx < len(nums2)):
                nums3.append(nums2[jdx])
                jdx = jdx + 1
        if jdx == len(nums2) and idx < len(nums1):
            while(idx < len(nums1)):
                nums3.append(nums1[idx])
                idx = idx + 1
        if len(nums3) % 2 == 1:
            return float(nums3[len(nums3) / 2])
        else:
            return float(nums3[len(nums3) / 2] + nums3[len(nums3) / 2 - 1]) / 2.0

算法性能：
在这里插入图片描述

但是算法的时间复杂度为 $o (m + n)$ ，而题目要求是时间复杂度 $o (l o g (m + n))$ ,因此考虑用二分查找：


class Solution(object):
    def findMedianSortedArrays(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: float
        """
        if (len(nums1) + len(nums2)) % 2 == 1:
            return float(self.findMaxK(nums1, nums2, (len(nums1) + len(nums2)) / 2 + 1))
        else:
            return float((self.findMaxK(nums1, nums2, (len(nums1) + len(nums2)) / 2 ) + self.findMaxK(nums1, nums2, (len(nums1) + len(nums2)) / 2 + 1)) / 2.0)

    def findMaxK(self, nums1, nums2, k):
        if len(nums1) > len(nums2):
            nums1, nums2 = nums2, nums1
        if len(nums1) == 0:
            return nums2[k - 1]
        if k == 1:
            if nums1[0] < nums2[0]:
                return nums1[0]
            else:
                 return nums2[0]
        idx = min(k/2 - 1, len(nums1) - 1)
        if nums1[idx] <= nums2[k - idx - 2]:
            return self.findMaxK(nums1[idx + 1 :], nums2, k - idx - 1)
        else:
            return self.findMaxK(nums1, nums2[k - idx - 1 :], idx + 1)

在这里插入图片描述