【刷算法】LeetCode.4-寻找两个有序数组的中位数

509 阅读1分钟

给定两个大小为 m 和 n 的有序数组 nums1nums2

请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。

你可以假设 nums1nums2 不会同时为空。

示例 1:

nums1 = [1, 3]
nums2 = [2]

则中位数是 2.0

示例 2:

nums1 = [1, 2]
nums2 = [3, 4]

则中位数是 (2 + 3)/2 = 2.5

解答

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number}
 */
var findMedianSortedArrays = function(nums1, nums2) {
  nums1.push(...nums2);
  nums1.sort((a, b) => {
    return a-b;
  });
  
  let len = nums1.length;
  if(len%2 === 1) {
    let mid = (len-1)/2+1-1;
    return nums1[mid];
  }else{
    let mid1 = len/2-1;
    let mid2 =  mid1+1;
    return (nums1[mid1]+nums1[mid2])/2;
  }
};