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

Github地址

思路

条件①两个数组都是有序②时间复杂度要求O(log(m+n))

• 在一个有序数组中计算中位数很方便,但这是两个, 虽然是两个,但是有序,所以两个数组看成一个就行
• 但是要考虑到一个数组还有很长,另一个数组还有元素
• 最后函数返回值为float

编写

class Solution:

    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        # 设置两个索引, index1, index2分别跟随nums1, nums2
        # first,second 代表第一大和第二大
        first = second = 0
        index1 = index2 = 0
        len1 = len(nums1)
        len2= len(nums2)
        mid = (len1+len2) //2 +1
        for i in range(mid):
            second = first
            if( (index2 >= len2) or ((index1 < len1) and (nums1[index1] <= nums2[index2]))):
                first = nums1[index1]
                index1 += 1
            else:
                first = nums2[index2]
                index2 += 1
        
        if((len1+len2) % 2 == 0):
            return (first+second)/2
        else:
            return float(first)

''' #快速解法   
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        nums = sorted(nums1 + nums2)
        mid = len(nums)//2
        if(len(nums)%2 == 0):
            return (nums[mid] + nums[mid-1])/2
        else:
            return float(nums[mid])
'''

思考与学习