题目描述
给定两个大小分别为 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
思路分析
由于O(log (m+n))意味着我们需要采用类似二分法的思路,
中位数位置计算规则如下:数据个数是奇数时,中位数是下标为arr[len/2],当数据个数是偶数时,中位数是下标为(arr[len/2] + arr[len/2-1]) / 2
算法:
- 计算出中位数的下标,然后循环到下标的位置即可
- 数组nums1、nums2分别从头开始取,谁小取谁,直到退出循环
- 循环过程中,记录当前值和前一个值
- 循环结束后,奇数时,返回当前取值,偶数时,返回当前值、前一个值的和 再除以2
- 该算法并非最佳算法,但是可以符合要求,也比较简单
算法
var findMedianSortedArrays = function(nums1, nums2) {
let len = nums1.length + nums2.length;
// 数据个数是奇数时mid = len / 2,个数是偶数时mid = len / 2 + 1
let mid = Math.floor(len / 2);
let j = 0, k = 0, result = 0, result1 = 0;
for(let i= 0; i <= mid; i++) {
if (k >= nums1.length) {
result1 = result;
result = nums2[j];
j++;
continue;
}
if (j >= nums2.length) {
result1 = result;
result = nums1[k];
k++;
continue;
}
if (nums1[k] < nums2[j]) {
result1 = result;
result = nums1[k];
k++;
} else {
result1 = result;
result = nums2[j];
j++;
}
}
return len % 2 === 0 ? (result1 + result) / 2 : result;
};
PS:leetcode心得,坚持刷,欢迎交流