寻找两个正序数组的中位数

DESC:

给定两个大小分别为 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(log (m+n)) 的算法解决此问题吗？

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/median-of-two-sorted-arrays
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

CODE:

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        if (nums1.length > nums2.length) {
            return this.findMedianSortedArrays(nums2, nums1);
        }
        int m = nums1.length;
        int n = nums2.length;
        //int total = (m+n+1)/2;
        int total = m+(n-m+1)/2;//重点
        int left=0, right=m;
        while (left<right) {//重点
            int i = left + (right-left+1)/2;//重点
            int j = total -i;
            if (nums1[i-1] > nums2[j]) {
                right = i-1;
            } else {
                left = i;//重点
            }
        }
        int i=left, j=total-i;
        int nums1Left = i == 0 ? Integer.MIN_VALUE:nums1[i-1];
        int nums1Right = i == m ? Integer.MAX_VALUE:nums1[i];
        int nums2Left = j == 0 ? Integer.MIN_VALUE:nums2[j-1];
        int nums2Right = j == n ? Integer.MAX_VALUE:nums2[j];

        return (m+n)%2==0 ? (double)(Math.max(nums1Left, nums2Left)+Math.min(nums1Right, nums2Right))/2 : Math.max(nums1Left, nums2Left);
    }
}

Python

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        m = len(nums1)
        n = len(nums2)
        if m>n:
            return self.findMedianSortedArrays(nums2, nums1)
        
        total = m+(n-m+1)//2 #重点
        left=0
        right=m
        while left<right:
            i= left+(right-left+1)//2 #重点
            j=total-i
            if nums1[i-1]>nums2[j]:
                right=i-1
            else:
                left=i #重点
        i = left
        j = total-i
        nums1Left = -inf if i==0 else nums1[i-1] #重点
        nums1Right = inf if i==m else nums1[i]
        nums2Left = -inf if j==0 else nums2[j-1]
        nums2Right = inf if j==n else nums2[j]
        return (float) (max(nums1Left, nums2Left)+min(nums1Right, nums2Right))/2 if (m+n)%2==0 else max(nums1Left, nums2Left)

NOTES:

1. 要求时间复杂度尽可能小，该方法时间复杂度为O(log(min(m,n)))，空间复杂度为：O(1)。
2. 二分法，基于两个条件去查找将两个数组分割为左右两部分的分割线，条件一：分割后，两数组左右部分对应相加后元素个数相等或差1（左边比右边多一个），取决于总元素个数是奇数偶数；条件二：左侧区域元素均小于右侧元素。
3. 
4. 为了兼容奇数偶数情况，左侧元素个数统一使用公式：leftCount=(m+n+1)/2，所以如果数组一分割点为i-1,i,i也代表了数组一左侧元素的个数，所以数组二分割点为j=leftCount-i, j-1;