我的刷题日记（25）

给定两个大小为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的中位数。

var findMedianSortedArrays = function(nums1, nums2) {
    // 根据两个数组的总长度 判断是奇数还是偶数
    let midIndex1;  // 第一个中位数的位置
    let midIndex2; // 第二个中位数的位置
    let len = nums1.length+nums2.length;
   if (len%2 === 0 && len > 2) {
        // 偶数个数的中位数有两个
        midIndex1 = len/2-1;
        midIndex2 = midIndex1+1;
    } else if (len%2 !== 0 && len > 2) {
        // 奇数个数的中位数只有一个
        midIndex1 = (len+1)/2 -1
    } else if (len === 2) {
        // 中位数为两数的平均数
        const arr = [...nums1, ...nums2];
        return (arr[0]+arr[1])/2
    } else if (len === 1) {
        // 只有一个数时的中位数
        return [...nums1, nums2][0]
    } else {
        return 0;
    }

    const resArr = [];
    while (nums1.length > 0 || nums2.length > 0) {
        //比较两个数组的首元素大小，小者 shift 出来 push 到结果数组中去
        if (nums1.length === 0) {
            resArr.push(...nums2)
            break;
        }
        if (nums2.length === 0) {
            resArr.push(...nums1);
            break;
        }
        if (nums1[0] < nums2[0]) {
            resArr.push(nums1.shift())
        } else if (nums1[0] > nums2[0]) {
            resArr.push(nums2.shift())
        } else {
            resArr.push(nums1.shift())
            resArr.push(nums2.shift())
        }
        // 刚好到中位数的位置，直接返回
        if (midIndex2 !== undefined && resArr.length-1 === midIndex2) {
            return (resArr[midIndex1] +resArr[midIndex2])/2;
        } else if (midIndex2 === undefined && resArr.length-1 === midIndex1) {
            return resArr[midIndex1]
        }
    }
    if (midIndex2 !== undefined) {
    	// 偶数个数的中位数
        return (resArr[midIndex1] +resArr[midIndex2])/2;
    } else {
    	// 奇数个数的中位数
        return resArr[midIndex1]
    }
};