JS Li Kou の古典的な 100 の質問 - 2 つの正の配列配列の中央値を求めます

4. 2 つの順序配列の中央値を求める

サイズと の 2 つの正の順序 (m小さいものからn大きいものへ) の配列が 与えられます。これら 2 つの順序配列の中央値を見つけて返してください。nums1nums2

アルゴリズムの時間計算量は でなければなりません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

解法：没有达到预期的时间复杂度，懒人解法，将数组合并为一个有序数组，根据数组长度返回中位数

/**
 * @param {number[]} nums1
 * @param {number[]} nums2
 * @return {number}
 */
 var findMedianSortedArrays = function(nums1, nums2) {
    let arr = []
    let i=0, j=0, m=0
    while(i<nums1.length || j<nums2.length) {
        if(i<nums1.length && j<nums2.length) {
            arr[m] = nums1[i]<= nums2[j] ? nums1[i++] : nums2[j++]
            m++
        } else if(i<nums1.length){
            arr[m++] = nums1[i++]
        } else if(j<nums2.length){
            arr[m++] = nums2[j++]
        }
    }
    let mid = nums1.length + nums2.length
    if(mid%2 !=0) {
        let a =Math.floor(mid/2)
        return arr[a]
    } else {
        return (arr[mid/2-1]+arr[mid/2])/2
    }
};