4. Median of Two Sorted Arrays

最新推荐文章于 2024-04-05 16:34:21 发布

xiaoran_zhu

最新推荐文章于 2024-04-05 16:34:21 发布

阅读量297

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode

本文链接：https://blog.youkuaiyun.com/xiaoran_zhu/article/details/50896848

Leetcode 专栏收录该内容

7 篇文章

订阅专栏

本文介绍了一种新颖的方法来寻找两个已排序数组的中位数，避免了常见的二分查找过程。通过一次性合并两个数组并遍历一次即可得出结果，提高了效率。

没有用很多博客提到的A,B两个数组每次砍一半的方法，直接归并后取值，遍历一次数组即可。

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number}
 */
var findMedianSortedArrays = function(nums1, nums2) {
    var res_array = merge(nums1,nums2),
        nums1_len = nums1.length,
        nums2_len = nums2.length,
        i = parseInt( (nums1_len+nums2_len-1)/2 ),
        j = parseInt( (nums1_len+nums2_len)/2   );
        
    return (res_array[i] + res_array[j]) / 2; //返回中间那个数
    
};

var merge = function(nums1, nums2){
    var ret = [],
        nums1_len = nums1.length,
        nums2_len = nums2.length,
        i = 0 ,
        j = 0 ;
    
    //归并两个数组，直到一个到头
    while(i<nums1_len && j<nums2_len){
        if(nums1[i] < nums2[j] ){
            ret.push(nums1[i]);
            i++;
        }
        else{
            ret.push(nums2[j]);
            j++;
        }
    }
    
    //剩下的那个数组继续插入
    if(i<nums1_len){
        for(i;i<nums1_len;i++){
            ret.push(nums1[i]);
        }
    }
    
    if(j<nums2_len){
        for(j;j<nums2_len;j++){
            ret.push(nums2[j]);
        }
    }
    //返回值
    return ret;
};