【LeetCode热题100(66/100)】寻找两个正序数组的中位数

题目地址:链接

思路: 要想O(log(n + m))时间复杂度内实现当前算法。很显然是二分算法,而且需要将两个数组成为一个整体来想。
两个数组都是递增的,并且中位数的左边的内容是一样的,如果将问题转换为获取中位数的下标,那么思路就很清晰了。

  1. 获取两个数组下标。在 nums1 中选取一个下标 mid - 1,由于是两个数组的中位数,而且是两个有序数组,所以另外一个数组的下标是确定的totalN - mid - 1
  2. 选取两个下标的最大值,判断是否大于nums[q1 + 1],如果成立,说明当前成立,可以让左收拢;反之向右收拢。
  3. 重复1,知道 l < r 不成立
  4. 根据 n + m 的奇偶性来返回对应的中位数。
/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number}
 */

var findMedianSortedArrays = function(nums1, nums2) {
    function check_left(mid) {
        if(mid > n) return false;
        if(totalN - mid > m) return true;
        let [q1, q2] = [mid - 1, totalN - mid - 1];
        let maxValue = -Infinity;
        if(q1 >= 0) maxValue = Math.max(nums1[q1], maxValue);
        if(q2 >= 0) maxValue = Math.max(nums2[q2], maxValue);
        if(q1 + 1 < n && maxValue > nums1[q1 + 1]) return true;
        return false;
    }
    let [n, m] = [nums1.length, nums2.length];
    let totalN = n + m + 1 >> 1;
    let isOdd = (n + m) % 2;

    let [l, r] = [0, totalN];
    while(l < r) {
        let mid = l + r >> 1;
        if(check_left(mid)) l = mid + 1;
        else r = mid;
    }

    let ans = -Infinity;
    let [idx1, idx2] = [l - 1, totalN - l - 1]
    if(idx1 >= 0) ans = Math.max(ans, nums1[idx1]);
    if(idx2 >= 0) ans = Math.max(ans, nums2[idx2]);
    if(!isOdd) {
        let otherNum = Infinity
        if(l < n) {
            otherNum = Math.min(otherNum, nums1[l]);
        }
        if(totalN - l < m) {
            otherNum = Math.min(otherNum, nums2[totalN - l]);
        }
        ans = (ans + otherNum) / 2;
    }
    return ans;
};
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值