[路飞]_leetcode-4-寻找两个正序数组的中位数

137 阅读1分钟

题目描述

[题目地址]

给定两个大小分别为 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

示例 3:

输入: nums1 = [0,0], nums2 = [0,0]
输出: 0.00000

示例 4:

输入: nums1 = [], nums2 = [1]
输出: 1.00000

示例 5:

输入: nums1 = [2], nums2 = []
输出: 2.00000

提示:

  • nums1.length == m
  • nums2.length == n
  • 0 <= m <= 1000
  • 0 <= n <= 1000
  • 1 <= m + n <= 2000
  • -106 <= nums1[i], nums2[i] <= 106

通过次数552,171提交次数1,347,470

本题需要知道的是如果输入两个数组的长度之和 len为奇数,则中位数为中间一个元素(下标为 len>>1 )的值,否则为中间两个元素 (下标为 len>>1len>>1+1) 的平均值)

本题解题并不难,难点在于题解的时间复杂度应该为 O(log (m+n)) 

首先我们看一下时间复杂度没达到给定要求但是比较好理解的题解 => 题解1 & 题解2

代码如下:

var findMedianSortedArrays = function(nums1, nums2) {
  // 合并数组
  const arr = [...nums1,...nums2],
  // 获取数组长度
  len = arr.length;
  // sort 排序
  arr.sort((a,b) => a-b)
  // 获取中间下标
  const midIndex = len>>1
  // 判断数组长度奇偶返回对应结果
  return len%2?arr[midIndex]:(arr[midIndex-1]+arr[midIndex])/2
}

至此就完成了leetcode-4-寻找两个正序数组的中位数