LeetCode4. 寻找两个有序数组的中位数

题目描述

寻找两个有序数组的中位数
给定两个大小为 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

解决方法

方法一：合并，排序，除以2

对于一个有序数组，如果数组长度是奇数，那么中位数就是中间那个值，如果长度是偶数，就是中间两个数的平均数。

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

Time Complexity: $O(n)$
大于题目要求的时间复杂度

方法二：寻找划分位置

中位数的作用：将一个集合划分为两个长度相等的子集，其中一个子集中的元素总是大于另一个子集中的元素。

思路：nums1和nums2分别以i,j为界划分成左右两部分，不断调整i,j,以达成以下条件：

1. len(nums1_左) + len(nums2_左) = len(nums1_右) + len(nums2_右) 就是说，划分后的左右两边的长度相等。
2. Max(nums1左、nums2左) ≤ Min(nums1右、nums2右) 就是说，左边的元素总是小于右边。

class Solution:
    def findMedianSortedArrays(self, nums1, nums2):
        """
        :type nums1: List[int]
        :type nums2: List[int]
        :rtype: float
        """
        m, n = len(nums1), len(nums2)
        if m > n:
            nums1, nums2, m, n = nums2, nums1, n, m
        if n == 0:
            raise ValueError
            
        imin, imax, half_len = 0, m, (m+n+1)//2
        while imin <= imax:
            i = (imin + imax)//2
            j = half_len - i
            if i<m and nums2[j-1] > nums1[i]: #i太小
                imin = i+1
            elif i>0 and nums1[i-1]>nums2[j]: #i太大
                imax = i-1
            else:#找到i的正确位置
                if i==0: max_of_left = nums2[j-1]#临界问题，其中一个列表全在右边，左边为空
                elif j==0: max_of_left = nums1[i-1]
                else: max_of_left = max(nums1[i-1], nums2[j-1])
                    
                if(m+n)%2 == 1: #若列表长度为奇数
                    return max_of_left
                
                if i==m: min_of_right = nums2[j] #临界问题，其中一个列表全在左边，右边为空
                elif j==n: min_of_right = nums1[i]
                else: min_of_right = min(nums1[i], nums2[j])
                    
                return (max_of_left + min_of_right)/2

Time Complexity: $O(log(min(m,n)))$
首先，查找的区间是 [0,m]。 而该区间的长度在每次循环之后都会减少为原来的一半。 所以，我们只需要执行 \log(m) 次循环。由于我们在每次循环中进行常量次数的操作，所以时间复杂度为 $O(log(m))$。 由于 $m\le n$，所以时间复杂度是 $O(\log (\min (m,n)))$ 。
空间复杂度：$O(1)$， 我们只需要恒定的内存来存储 9 个局部变量， 所以空间复杂度为 $O(1)$。

参考

• 力扣官方题解