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) O(m+n)
-
空间复杂度: O ( m + n ) 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
};