【NO.4】LeetCode HOT 100—4. 寻找两个正序数组的中位数

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

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] <= 10^6

解题

方法：二分查找（变体）

两个数组长度确定，那我们就可以确认中位数的具体值，计作第k个

那我们分别从两个数组中找到前 k/2 个 数，谁的数小，就把谁排除，因为我们需要的是大的那个数

一直减下去，直到k=1，此时，对比两个数组的第一个数，谁小谁就是我们要找的数，

这个过程就是将前k-1个数给排除掉，然后剩下的第一个就是我们要找的数

因为有两个数组，这个过程要注意下标越界的问题，如果有一个数组越界了，说明这个数组的数全排除完了，

那就只看另外一个数组就好了

// 时间log（n+m），空间1
class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int m = nums1.length;
        int n = nums2.length;
        // 这里有个小技巧，求中位数要根据奇偶数来找，有个公式，可以忽略奇偶数的问题
        // 假设长度为 n，求中位数k = ((n+1)/2 + (n+2)/2) /2
        int left = (n+m+1)/2;
        int right = (n+m+2)/2;
        // 找到第left 第right个数
        return (findNum(nums1, 0, nums2, 0, left) + findNum(nums1, 0, nums2, 0, right)) / 2.0;
    }

    // i,j代表数组的下标， k代表找到第k个元素
    private int findNum (int[] nums1, int i, int[] nums2, int j, int k) {
        // nums1数组找完了，就找nums2数组中第k个数
        if (i >= nums1.length) {
            return nums2[j + k - 1];
        }
        if (j >= nums2.length) {
            return nums1[i + k - 1];
        }

        if (k == 1) {
            return nums1[i] < nums2[j]? nums1[i] : nums2[j];
        }

        // 分别从两个数组中找到前 k/2 个 数，谁的数小，就把谁排除
        int num1 = (i + k/2 - 1) < nums1.length ? nums1[i + k/2 - 1] : Integer.MAX_VALUE;
        int num2 = (j + k/2 - 1) < nums2.length ? nums2[j + k/2 - 1] : Integer.MAX_VALUE;
        if (num1 > num2) {
            // 将小的数组前k/2个排除掉，那下标就是从num1 / num2的下一位开始，找剩下第k-k/2个数
            return findNum(nums1, i, nums2, j + k/2, k-k/2);
        } else {
            return findNum(nums1, i + k/2, nums2, j, k-k/2);
        }
    }
 


}