00二分查找困难 LeetCode4. 寻找两个正序数组的中位数

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

描述

给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
算法的时间复杂度应该为 O(log (m+n)) 。

分析

暴力方法是合并两个数组，但是时间复杂度就变成了O(m+n)了。
O(log(m+n))的思路是，求两个有序数组的第k小元素，本地k就是中间的数(或中间的平均值)。
针对求两个有序数组的第k小，可以理解为，对两个数组分别求第k/2小的数，比较大小，较小的数及其前面的元素是不可能成为第k小的数，所以把这些数排除掉。这样每次都能排除掉一个数组现存元素一半的元素。（像二分查找每次都能排除掉一半）
忽略奇偶性，使用公式（(m + n + 1) / 2 + (m + n + 2) / 2）/ 2 可以得到最中间的那个数，无论原数组数量是奇数还是偶数。
注意k/2有可能会越界，需要做越界的判断。

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int m = nums1.length;
        int n = nums2.length;
        int left = (m + n + 1) / 2;
        int right = (m + n + 2) / 2;
        return (recur(nums1,0,nums2,0,left) + recur(nums1,0,nums2,0,right)) / 2.0;
    }

    public int recur(int[] nums1, int start1, int[] nums2, int start2, int k) {
        if (start1 == nums1.length) {
            return nums2[start2+ k - 1];
        }
        if (start2 == nums2.length) {
            return nums1[start1+ k - 1];
        }
        if (k == 1) {
            return Math.min(nums2[start2],nums1[start1]);
        }
        int index1 = start1 + k / 2 - 1;
        index1 = Math.min(nums1.length-1,index1);
        int index2 = start2 + k / 2 - 1;
        index2 = Math.min(nums2.length-1,index2);
        if (nums1[index1] < nums2[index2]) {
            return recur(nums1,index1+1,nums2,start2,k-(index1-start1+1));
        } 
        return recur(nums1,start1,nums2,index2+1,k-(index2-start2+1));
    }
}