leetcode 寻找两个正序数组的中位数

最新推荐文章于 2024-07-07 11:57:48 发布

Donnx

最新推荐文章于 2024-07-07 11:57:48 发布

阅读量118

点赞数

文章标签： leetcode 算法

本文链接：https://blog.csdn.net/Donnx/article/details/127842502

版权

记录[B站视频]以及leetcode官方解答(https://www.bilibili.com/video/BV1H5411c7oC/?spm_id_from=333.337.search-card.all.click&vd_source=0276edd699c49c1ca641d9a0d0ba0503)
思路：
划分边界，设nums1.length == m < n == nums2.length
为什么要确保 m < n 呢，因为后续要用到 j = (m + n + 1) / 2 - i，我们是根据 i 求 j，如果 m > n ,求出的j可能为负数。
接下来将nums1数组化为左边 i 个数[0 ,i-1]，num2数组化为左边 j 个数[0 , j - 1]
要确保如下条件：
①对于m + n 为偶数的情况 i + j = (m + n) / 2
②对于 m + n 为奇数的情况， i + j = (m + n + 1) / 2
但加不加1不影响结果，所以 j = (m + n + 1) / 2 - i
主要的问题就是确定 i 的值，i确定了j随之也就确定了
i 要保证B[j - 1] < A[i] 且 A[i-1] < B[j]
即i 和 j左边的数要小于右边的数
采用二分法确定i
如果j != 0,i != m,且B[j-1] > A[i] 则i往右边移动
如果i != 0, j != n且A[i-1] > B[j] 则i往左移

如果i = 0，则左边最大值为B[j-1]
如果j=0,则左边最大值为A[i-1]
否则左边最大值为max(A[i-1],B[j-1])

判断数组长度是否为奇数
奇数：返回左边最大值
偶数：还要求右边最小值

如果i = m 右边最小值为B[j],
如果j = n右边最小值为A[i]
返回左最大和右最小的平均数

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 的区间
        int iMin = 0, iMax = m;
        while(iMin <= iMax){
            int i = (iMin + iMax) / 2;
            int j = (m + n + 1) / 2 - i;
            if(j != 0 && i != m && 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]);
                }
                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 + minRight) / 2.0;
            }
        }
        return -1;
    }
        // 参考https://www.bilibili.com/video/BV1H5411c7oC/?spm_id_from=333.337.search-card.all.click&vd_source=0276edd699c49c1ca641d9a0d0ba0503
}```