No.4寻找两个有序数组的中位数-CSDN博客

本文链接：https://blog.csdn.net/qingxuanmingye/article/details/89035542

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

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

本题难点：算法复杂度有限制（log(n)）。不能使用简单的列表合并再排序。给出的解法是，新建列表。比较两个列表。较小的放入。注意新列表的初始化为0.还有列表一个为空后。另一个的判断。把后续放入列表。还有就是注意。求中位数的时候，考虑到列表坐标从零开始。需要len（）减一。还要注意最后输出的是浮点型。//和/的区别。前一个整除。后一个float型。

解题：

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        #创建新列表，并将所有元素初始化为0
        nums3 = [0] * (len(nums1)+len(nums2))
        l_i,r_i,i = 0,0,0
        while(l_i<len(nums1)) and (r_i<len(nums2)):
            if nums1[l_i] < nums2[r_i]:
                nums3[i] = nums1[l_i]
                l_i += 1
            else:
                nums3[i] = nums2[r_i]
                r_i += 1
            i += 1
        if l_i != len(nums1):
            nums3[i:] = nums1[l_i:]
        else:
            nums3[i:] = nums2[r_i:]
        len_3 = len(nums3)
        if len_3 % 2 != 0 :
            return float(nums3[(len_3-1)//2])
        return (nums3[(len_3)//2-1]+nums3[(len_3)//2])/2