两个有序数组的中位数——划分数组法

两个有序数组的中位数（LeetCode 04）

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

        //第一步: 初始化i的区间，i的最小值为0，最大值为m
        //num1数组的索引是从0开始，如 1 2 2 3 4 5 ，
        //而i的取值就可以理解成划分数组的空白处，即从第一个数值前面的空白到最后一个数值后的空白，共有m+1个位置（取值从0开始，末尾就是m）
        int iMin = 0, iMax = m;
        while(iMin <= iMax) {
            int i = iMin + (iMax - iMin) / 2;//二分
            int j = (m + n + 1) / 2 - i;//若m+n是偶数就是(m+n)/2，而偶数的前提下(m+n)/2 = (m+n+1)/2；若m+n为奇数就是(m+n+1)/2

            //第二步: 交叉判断i是否合理
            if(i != m && j != 0 && nums2[j - 1] > nums1[i]) {
                //左下大于右上，i偏小
                iMin = i + 1;
            } else if(i != 0 && j != n && nums1[i - 1] > nums2[j]) {
                //左上大于右下，i偏大
                iMax = i - 1;
            } else {
                //第三步: 此时i是合理的，将边界条件拿出来判断
                int maxLeft = 0;
                if(i == 0) maxLeft = nums2[j - 1];//此时没有左上，就直接取左下即可
                else if(j == 0) maxLeft = nums1[i - 1];//此时没有左下，就直接取左上
                else maxLeft = Math.max(nums1[i - 1], nums2[j - 1]);//左上和左下都有，取两者的最大值

                //第四步: 判断m+n是否为偶数，根据奇偶情况有不同的输出计算
                if((m + n) % 2 == 1) return maxLeft;//奇数情况
                //偶数情况需要再求右边的最小值
                int minRight = 0;
                if(i == m) minRight = nums2[j];//此时没有右上，直接取右下
                else if(j == n) minRight = nums1[i];//此时没有右下，就直接取右上
                else minRight = Math.min(nums1[i], nums2[j]);//右上和右下都有，取两者的最小值
                return maxLeft + (double)(minRight - maxLeft) / 2;
            }
        }
        return 0;//实际上不会执行到这里呀
    }
}