LeetCode | 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

题解

本题所给条件是两个有序数组，要寻找其中位数。那么，将两个数组合并在一起，找一下中位数就好了。因为已经给出两数组是有序的（从小到大），所以只需要循环比较两数组上的值，依次将小的值放到心得数组当中就好了。然后判断合并后的数组长度，再确定直接取中间值还是取中间两数求平均。

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        l1 = l2 = l3 = 0
        nums3 = [0] * len(nums1 + nums2)
        while l1 < len(nums1) and l2 < len(nums2):
            if nums1[l1] < nums2[l2]:
                nums3[l3] = nums1[l1]
                l1 += 1
            else:
                nums3[l3] = nums2[l2]
                l2 += 1
            l3 += 1
        if l1 == len(nums1):
            nums3[l3:] = nums2[l2:]
        else:
            nums3[l3:] = nums1[l1:]
        if len(nums3) % 2 ==0:
            return (nums3[len(nums3)//2-1] + nums3[len(nums3)//2]) / 2
        else:
            return nums3[len(nums3)//2]

上述的程序完全是最基本的，既然用python来解决，就要利用好python语言的特点。首先，python中的列表有extend方法，直接将一个数组添加到另一个数组后面，这样可以减小内存消耗。同时python还有sort方法，可以将列表中的数字排序，而且经过实践，发现python内置的sort方法的排序速度也是很快的，有兴趣的朋友可以查看其源码仔细了解一下。

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        nums1.extend(nums2)
        nums1 = sorted(nums1)
        if len(nums1) % 2 ==0:
            return (nums1[len(nums1)//2-1] + nums1[len(nums1)//2]) / 2
        else:
            return nums1[len(nums1)//2]

扫描二维码

获取更多精彩

小田学Python

       

       

                 文章好看就点这里