【LeetCode 4】寻找两个有序数组的中位数

1.题目描述

给定两个大小为 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.思路：

因为两个数组是有序的，所以可以先将两个数组合并成一个数组，再计算中位数。与‘合并两个有序的链表’类似，不过这题还要返回合并后的数组的中位数。

• 首先初始化一个栈stack，用来存放合并后的数；然后初始化两个指针left1和left2，分别指向nums1和nums2的起始位置；
• 开始遍历，比较nums1[left1]和nums2[left2]的大小，若nums1[left1]小一些，则将nums1[left1]追加到stack中，left1指针右移一个，继续与nums2[left2]进行比较；同理若nums2[left2]小一些，则将nums2[left2]追加到stack中，left2指针右移一个，继续与nums1[left1]进行比较；
• 跳出循环无非两种情况，一是nums1先遍历完了，二是nums2先遍历完了。若nums1先遍历完，则直接执行stack += nums2[left2:]，加上nums2数组剩下的部分即可；若nums2先遍历完，同理；
• 最后判断合并后得到的stack的长度是奇数还是偶数，若是奇数，则直接返回中间值即为中位数；若为偶数，则需要将中间两个数相加再除以2，才是中位数。

3.Python代码实现

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        m, n = len(nums1), len(nums2)
        if m == 0 and n == 0:
            return 0
        stack = []
        left1, left2 = 0, 0
        while left1 < m and left2 < n:
            if nums1[left1] <= nums2[left2]:
                stack.append(nums1[left1])
                left1 += 1
            else:
                stack.append(nums2[left2])
                left2 += 1
        if left1 == m:
            stack += nums2[left2:]
        elif left2 == n:
            stack += nums1[left1:]

        s = len(stack)
        if s % 2 == 1:
            return stack[s // 2]
        else:
            return (stack[s//2] + stack[s//2 - 1]) / 2