LeetCode 第 4 号问题:寻找两个正序数组的中位数

LeetCode 第 4 号问题:寻找两个正序数组的中位数

题目地址

https://leetcode-cn.com/problems/median-of-two-sorted-arrays/

题目描述

给定两个大小为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。

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

你可以假设 nums1 和 nums2 不会同时为空。

示例 1:

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

则中位数是 2.0
示例 2:

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

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

思路

暴力解决方法:拼接后找中位数

代码一

const findMedianSortedArrays = function(nums1, nums2) {
    let arr=nums1.concat(nums2)// 合并两个数组
    arr.sort((a, b) => a - b)// 新数组从小到大排序
    let length = arr.length
    if(length%2==0){
        /**
         * 数组长度为偶数,输出中间两数之和的平均值
        */
        return (arr[length/2]+arr[length/2-1])/2
    }else{
        /**
         * 数组长度为奇数,输出中间数
        */
        return arr[(length-1)/2]
    }
};

代码二

代码最少的方法,时间复杂度为O((m + n)log(m + n))

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number}
 */
const findMedianSortedArrays = function(nums1, nums2) {
    const arr = [...nums1, ...nums2].sort((a, b) => a - b);
    const { length } = arr;
    return length % 2 ? arr[Math.floor(length / 2)] : (arr[length / 2] + arr[length / 2 - 1]) / 2;
}

LeetCode 第 4 号问题:寻找两个正序数组的中位数

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章