力扣刷题之4.寻找两个正序数组的中位数

仅做学习笔记之用。

题目：

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

算法的时间复杂度应该为 O(log (m+n)) 。

示例 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

提示：

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

其实就是二分查找嘛。如果没了解过二分查找的可以看一下这篇文章

【数据结构与算法】——二分查找

        这段代码首先确保较短的数组是nums1，然后使用二分查找在nums1上寻找合适的分割点。在找到合适的分割点后，我们可以计算出中位数。如果两个数组的总长度是奇数，中位数就是左边部分的最大值；如果是偶数，中位数就是左边部分的最大值和右边部分的最小值的平均值。

代码如下：

class Solution {
    public static double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int m = nums1.length;
        int n = nums2.length;

        // 确保nums1是较短的数组
        if (m > n) { 
            int[] temp = nums1;
            nums1 = nums2;
            nums2 = temp;
            int tempLen = m;
            m = n;
            n = tempLen;
        }

        int Min = 0, Max = m, halfLen = (m + n + 1) / 2;
        while (Min <= Max) {
            int i = (Min + Max) / 2;
            int j = halfLen - i;

            if (i < m && nums2[j - 1] > nums1[i]) {
                Min = i + 1; // i太小
            } else if (i > 0 && nums1[i - 1] > nums2[j]) {
                Max = i - 1; // i太大
            } else {
                // 找到合适的i
                int maxLeft = (i == 0) ? nums2[j - 1] : 
                (j == 0) ? nums1[i - 1] : Math.max(nums1[i - 1], nums2[j - 1]);

                // 如果总长度是奇数，直接返回中位数
                if ((m + n) % 2 == 1) {
                    return maxLeft; 
                }

                int minRight = (i == m) ? nums2[j] : 
                (j == n) ? nums1[i] : Math.min(nums1[i], nums2[j]);

                // 如果总长度是偶数，返回中间两个数的平均值
                return (maxLeft + minRight) / 2.0; 
            }
        }
        return 0;
    }
}

（算法大佬英雄哪里出来曾曰：过——啦）

解决。