力扣hot100 寻找两个正序数组的中位数 二分排除法 满注释版 通俗易懂

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

思路

👨‍🏫 参考题解

• 二分排除法：寻找两数组第 $k$ 小的 x
• 两数组分别找到 $k/2$ 的位置，较小的值和它左边的值 就肯定比 x 小，排除掉 $y$ 个
• $k=k-y$：排除了 $y$ 个，寻找第 $k$小变成寻找第 $k-y$ 小
• 直到 $k=1$ 返回即可

💖 二分排除法

⏰ 时间复杂度: $O(\log (n\times m))$
🌎 空间复杂度: $O(1)$

class Solution {
	int n1, n2;

	public double findMedianSortedArrays(int[] nums1, int[] nums2)
	{
		n1 = nums1.length;
		n2 = nums2.length;
		int n = n1 + n2;
		if (n % 2 == 1)
		{
			int idx = (n + 1) / 2;
			return getKthElement(nums1, nums2, idx);
		} else// 偶数个元素，中位数为中间两个数的平均数
		{
			int minIdx = n / 2;
			int maxIdx = n / 2 + 1;
			double a = getKthElement(nums1, nums2, minIdx);
			double b = getKthElement(nums1, nums2, maxIdx);
			return (a + b) / 2.0;
		}
	}

	private double getKthElement(int[] nums1, int[] nums2, int k)
	{
		int idx1 = 0, idx2 = 0;// 表示两数组当前未被排除的数的起始下标
		while (true)
		{
			if (idx1 == n1)// nums1所有元素已经排除完了
				return nums2[idx2 + k - 1];
			if (idx2 == n2)// nums2所有元素已经排除完了
				return nums1[idx1 + k - 1];
			if (k == 1)
				return Math.min(nums1[idx1], nums2[idx2]);
			int half = k / 2;
			int newIdx1 = Math.min(idx1 + half, n1) - 1;
			int newIdx2 = Math.min(idx2 + half, n2) - 1;
			int x1 = nums1[newIdx1];
			int x2 = nums2[newIdx2];
			if (x1 <= x2)// 排除数组1 中 x1以及 x1左边的所有未排除元素
			{
				k -= (newIdx1 - idx1 + 1);// 排除了多少个， k 就减少多少个
				idx1 = newIdx1 + 1;// 更新未排除元素起始下标
			} else// 排除数组2 中 x2以及x2左边的所有未排除元素
			{
				k -= (newIdx2 - idx2 + 1);
				idx2 = newIdx2 + 1;// 更新未排除元素起始下标
			}
		}
	}
}

😋 数组切分法【TODO】

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
    	if(nums1.length > nums2.length) {
            return findMedianSortedArrays(nums2, nums1);
        }
        int m = nums1.length;
        int n = nums2.length;
        int begin = 0;
        int end = m;
        //左半部分的最大值
        double left_max = Double.MIN_VALUE;
        //右半部分的最小值
        double right_min = Double.MAX_VALUE;
        while(begin <= end) {
            //基于二分的方式求 i
            int i = begin + (end - begin) / 2;
            //数组A长度为m，数组B长度n，总长度为偶数时，左半部分右半部分相等：
            //m - i + n - j = i + j
            //总长度为奇数时，左半部分比右半部分多1个：
            //m - i + n - j + 1 = i + j
            //统一奇数、偶数情况，得到j为：(m + n + 1) / 2 - i
            int j = (m + n + 1) / 2 - i;
            //如果nums1[i - 1] > nums2[j]说明 i 取大了
            if(i > 0 && j < n && nums1[i - 1] > nums2[j]) {
                end = i - 1;
            }
            //nums2[j - 1] > nums1[i]，i 取小了
            else if(j > 0 && i < m && nums2[j - 1] > nums1[i]) {
                begin = i + 1;
            }
            //满足条件：nums1[i - 1] < nums2[j]，nums2[j - 1] < nums1[i]
            else {
                //边界情况，数组A切分后左半部分为空 i == 0
                //数组B 切分后左半部分为空 j == 0
                if(i == 0) {
                    left_max = nums2[j - 1];
                }
                else if(j == 0) {
                    left_max = nums1[i - 1];
                }
                //求左半部分的最大值
                else {
                    left_max = Math.max(nums1[i - 1], nums2[j - 1]);
                }
                //总长度为奇数时，直接返回左半部分最大值即可
                if((m + n) % 2 == 1) {
                    return left_max / 1.0;
                }
                //边界情况，数组A 切分后，右半部分为空 i == m
                //数组B 切分后，右半部分为空 j == n
                if(i == m) {
                    right_min = nums2[j];
                }
                else if(j == n) {
                    right_min = nums1[i];
                }
                //求右半部分的最小值
                else {
                    right_min = Math.min(nums1[i], nums2[j]);
                }
                //总长度为偶数时
                return (left_max + right_min) / 2.0;
            }
        }
        return 0.0D;
    }
}
/*
作者：王尼玛
链接：https://leetcode.cn/problems/median-of-two-sorted-arrays/solutions/600394/duo-tu-xiang-jie-liang-chong-shi-xian-by-75f4/
来源：力扣（LeetCode）
著作权归作者所有。商业转载请联系作者获得授权，非商业转载请注明出处。
*/