4. [✔][H]寻找两个正序数组的中位数
给定两个大小分别为 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
提示:
nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-106 <= nums1[i], nums2[i] <= 106
题解:
这个难点在于循环条件的指定 i <= Math.floor(len / 2)
,以及判断条件该选用nums1的值还是nums2的值if (l1 < m && (nums1[l1] < nums2[l2] || l2 >= n)) {
/*
* @lc app=leetcode.cn id=4 lang=typescript
*
* [4] 寻找两个正序数组的中位数
*/
// @lc code=start
function findMedianSortedArrays(nums1: number[], nums2: number[]): number {
let m = nums1.length;
let n = nums2.length;
let len = m + n;
let preValue = -1;
let curValue = -1;
let l1 = 0;
let l2 = 0;
for (let i = 0; i <= Math.floor(len / 2); i++) {
preValue = curValue;
// 为什么加l2 >= n?因为此时l2已经超出了n,nums2[l2] = undefined,如果不再上面添加这个条件
// l2会还往后+1,如果添加了,相当于结束为止在l1上
if (l1 < m && (nums1[l1] < nums2[l2] || l2 >= n)) {
curValue = nums1[l1];
l1++;
} else {
curValue = nums2[l2];
l2++;
}
}
return len % 2 === 0 ? (preValue + curValue) / 2 : curValue;
};
// @lc code=end