Leetcode JS/TS 4.寻找两个正序数组的中位数

2020-06-06  本文已影响0人  老北京程序员

给定两个大小为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。
请你找出这两个正序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。
你可以假设 nums1 和 nums2 不会同时为空。
示例 1:

nums1 = [1, 3]
nums2 = [2]

则中位数是 2.0
示例 2:

nums1 = [1, 2]
nums2 = [3, 4]

则中位数是 (2 + 3)/2 = 2.5

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number}
 */
var findMedianSortedArrays = function(nums1, nums2) {
    var n = nums1.length;
    var m = nums2.length;
    var left = Math.floor((n + m + 1) / 2)
    var right = Math.floor((n + m + 2) / 2)
    return (getKth(nums1, 0, n - 1, nums2, 0, m - 1, left) + getKth(nums1, 0, n - 1, nums2, 0, m - 1, right)) * 0.5;  
};

var getKth = function(nums1, start1, end1, nums2, start2, end2, k) {
    var len1 = end1 - start1 + 1;
    var len2 = end2 - start2 + 1;

    //让 len1 的长度小于 len2,这样就能保证如果有数组空了,一定是 len1 
    if (len1 > len2) {
        return getKth(nums2, start2, end2, nums1, start1, end1, k);
    }
    if (len1 == 0) {
        return nums2[start2 + k - 1];
    }

    if (k == 1){
        return Math.min(nums1[start1], nums2[start2]);
    }
    var i = start1 + Math.min(len1, Math.floor(k / 2)) - 1;
    var j = start2 + Math.min(len2, Math.floor(k / 2)) - 1;

    if (nums1[i] > nums2[j]) {
        return getKth(nums1, start1, end1, nums2, j + 1, end2, k - (j - start2 + 1));
    }
    else {
        return getKth(nums1, i + 1, end1, nums2, start2, end2, k - (i - start1 + 1));
    }

}


思路:
转化一下思路,找中位数也就是找合并后数组的第K个最小数,K为数组的长度的一半

————————
想要学习Cocos的同学,欢迎关注我的零基础Cocos教程
https://ke.qq.com/course/313749

上一篇 下一篇

猜你喜欢

热点阅读