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

163 阅读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

提示:

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

解题思路

  • 我们将数组合并后按升序排列
  • 如果是奇数返回最中间的数
  • 如果是偶数,返回中间两数的平均值

解题代码

var findMedianSortedArrays = function(nums1, nums2) {
    const arr = [...nums1,...nums2].sort((a,b)=>a-b)
    const len = arr.length
    return arr.length%2?arr[(len-1)/2]:(arr[len/2-1]+arr[len/2])/2
};

如有任何问题或建议,欢迎留言讨论!