寻找两个有序数组的中位数

不及你i

于 2022-01-20 19:55:16 发布

阅读量259

点赞数

文章标签：算法 leetcode

本文链接：https://blog.csdn.net/weixin_43730986/article/details/122608862

版权

个人主页：Sube’s blog

给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的中位数。

算法的时间复杂度应该为 O(log (m+n)) 。

示例 1：
输入：nums1 = [1,3], nums2 = [2]
输出：2.00000
解释：合并数组 = [1,2,3] ，中位数 2

示例 2：
输入：nums1 = [1,2], nums2 = [3,4]
输出：2.50000
解释：合并数组 = [1,2,3,4] ，中位数 (2 + 3) / 2 = 2.5

示例 3：
输入：nums1 = [0,0], nums2 = [0,0]
输出：0.00000

按照初中的思想，奇数个数，中位数就是中间的值，偶数个数就是中间的数加上右边的数的和的平均值
就…就完成了，并打败了90%的提交用户，不理解官方解释为啥这么长，可能我菜吧没理解出题人的初衷。
以下为代码，欢迎批评指正。
python

class Solution:
    def findMedianSortedArrays(self, nums1: List[int], nums2: List[int]) -> float:
        new = list(sorted(nums1 + nums2))
        mid = len(new) // 2
        return new[mid] if len(new) % 2 != 0 else (new[mid - 1] + new[mid]) / 2

javascript

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number}
 */
 var findMedianSortedArrays = function(nums1, nums2) {
    //  拼接两个数组
    nums1 = nums1.concat(nums2);
    // 排序，默认sort会转换成字符串再排序，如果出现负数将会出现错误
    // 所以我们采用自定义排序方法
    nums1.sort((a, b) => a - b);
    // 获取整型中位数
    mid = parseInt(nums1.length / 2);
    // 如果是奇数个数字，那么中位数即为mid
    if(nums1.length % 2 != 0){
        return nums1[mid];
    } else {
        // 否则为mid和mid-1个数和的平均值
        return (nums1[mid -1] + nums1[mid]) / 2;
    }
};

console.log(findMedianSortedArrays([3], [-2,-1]))