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

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

给定两个大小分别为 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

java代码：

 public static double findMedianSortedArrays(int[] nums1, int[] nums2) {
        if (nums1.length > nums2.length) {
            return findMedianSortedArrays(nums2, nums1);
        }

        int m = nums1.length;
        int n = nums2.length;
        //左边最大值
        int mid1 = 0;
        //右边最小值
        int mid2 = 0;
        int left = 0;
        int right = m;
        while (left <= right) {
            // 前一部分包含 nums1[0 .. i-1] 和 nums2[0 .. j-1]
            // 后一部分包含 nums1[i .. m-1] 和 nums2[j .. n-1]
            int i = (left + right) / 2;
            int j = (m + n + 1) / 2 - i;
            //考虑边界问题
            int n1_left = i == 0 ? Integer.MIN_VALUE : nums1[i - 1];
            int n1_right = i == m ? Integer.MAX_VALUE : nums1[i];
            int n2_left = j == 0 ? Integer.MIN_VALUE : nums2[j - 1];
            int n2_right = j == n ? Integer.MAX_VALUE : nums2[j];
            //存在一个最大的 i 满足A[i−1]≤B[j],这时候的A[i]>B[j−1]
            if (n1_left <= n2_right) {
                mid1 = Math.max(n1_left, n2_left);
                mid2 = Math.min(n1_right, n2_right);
                left = i + 1;
            } else {
                right = i - 1;
            }
        }

        return (m + n) % 2 == 0 ? (mid1 + mid2) / 2.0 : mid1;
    }