leetcode.4 寻找两个正序数组的中位数

力扣 4. 寻找两个正序数组的中位数https://leetcode-cn.com/problems/median-of-two-sorted-arrays/难度：hard

题目描述：

给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。

示例 1：

输入：nums1 = [1,3], nums2 = [2]
输出：2.00000
解释：合并数组 = [1,2,3] ，中位数 2
示例 2：

输入：nums1 = [1,2], nums2 = [3,4]
输出：2.50000
解释：合并数组 = [1,2,3,4] ，中位数 (2 + 3) / 2 = 2.5
示例 3：

输入：nums1 = [0,0], nums2 = [0,0]
输出：0.00000
示例 4：

输入：nums1 = [], nums2 = [1]
输出：1.00000
示例 5：

输入：nums1 = [2], nums2 = []
输出：2.00000
 

提示：

nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-106 <= nums1[i], nums2[i] <= 106

调试了好多次，终于对了。

空间复杂度 O(1)，时间复杂度 O(m+n)  m n分别为两个数组的长度

使用Python3，leetcode可以直接跑，代码及解释， 看下面吧：

#
# @lc app=leetcode.cn id=4 lang=python3
#
# [4] 寻找两个正序数组的中位数
#

# @lc code=start
class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        # 求一个数组的中位数
        def get_middle(nums, sum, target):
            res = 0        
            if len(nums) == 1:
                return nums[0]
            else:
                if sum % 2 != 0:
                    res = nums[target]
                else:
                    res = (nums[target-1]+nums[target])/2 
                return res 

        sum = len(nums1) + len(nums2)
        # 这里不用 target = int(sum/2) + 1, 是因为数组是从下表0开始的。
        target = int(sum/2) 
        i, j, N, res, tmp, flag = 0, 0, 0, 0, 0, False
        # 分情况讨论
        ## 情况1.有一个数组为空
        if len(nums1) == 0:
            res = get_middle(nums2, sum, target)
        elif len(nums2) == 0:
            res = get_middle(nums1, sum, target)
        ## 情况2.两个数组都不为空，逐个递进
        else: 
            # 此时，又分为两个情况，即两个数组和为偶数还是奇数
            if sum % 2 == 0:
                flag = True
            else: 
                flag = False 
            while N <= target: 
                x = nums1[i] if i<len(nums1) else sys.maxsize
                y = nums2[j] if j<len(nums2) else sys.maxsize
                tmp = min(x, y)
                if tmp == x:
                    i += 1 
                else:
                    j += 1    
                N += 1
                ## 和为偶数时，需要求两个数的和的均值，所以这里先把第一个数记录下来。
                    ### 下面条件为 N==target 而不是 N==target+1, 是因为逐个递进的时候，上面的N先加了1 
                if flag and N==target:
                    res += tmp
            res += tmp
        return res/2 if flag else res 


'''
    细节是魔鬼，前后调试了很多回，要命!!!
'''

# @lc code=end