JavaScript 实现寻找两个正序数组的中位数 ( LeetCode 热题 HOT 100 )

Problem: 4. 寻找两个正序数组的中位数

思路

思路很简单，就是先把两个数组合并，再重排一次顺序，然后合并后的数组长度是偶数个，就取中间两个和除以2，奇数个就取中间一个。这里有个规律，长度偶数时(要将中间两个数合起来才能看作中间值)，直接整除2，得到的就是数组中间两个中的前一个arr.length / 2 - 1，后一个则是arr.length / 2，注意索引对应的值要减一；长度奇数时，对长度加一，整除2得到的就是中间的一个(arr.length + 1) / 2 - 1，注意索引对应的值要减一。

解题方法

合并数组用的是 ES6 扩展运算符；排序没有手写实现，直接用 ES6 提供的数组排序API sort()；判断奇偶是看能否被2整除， num % 2 === 0余数为0，则能被整除。

复杂度

• 时间复杂度: $O(m+n)$

• 空间复杂度: $O(m+n)$

Code

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number}
 */
var findMedianSortedArrays = function(nums1, nums2) {
    let middle = 0
    const arr = [...nums1, ...nums2]
    if (arr.length === 0) {
        return 0
    }
    arr.sort((a, b) => a - b)
    const rest = arr.length % 2
    if (rest === 0) {
        const mid = arr.length / 2
        middle = Math.floor((arr[mid - 1] + arr[mid]) / 2 * 100000) / 100000
    } else {
        const mid = (arr.length + 1) / 2 - 1
        middle = arr[mid]
    }
    return middle
};