两个有序数组的中位数(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;//实际上不会执行到这里呀
}
}