4. 寻找两个正序数组的中位数

4. 寻找两个正序数组的中位数

原题

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

---

思路

边界特多，新手建议直接跳过

题解

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int len = nums1.size() + nums2.size();
        int mid = len / 2;
        if (len % 2 == 0) {
            //当长度为偶数时，需要中间两个数取平均值
            int lefght = find(nums1, 0, nums2, 0, mid), right = find(nums1, 0, nums2, 0, mid + 1);
            return (lefght + right) / 2.0;
        } else {
            //奇数取中间值,注意：mid是下标从0开始，但我们转化的问题是第k小的数，所以需要找到第mid+1个数
            return (find(nums1, 0, nums2, 0, mid + 1));    
        }
        
    }
	/**
		find函数：找到nums1从i开始，num2从j开始，两个数组中从小到大的第k小的数
	*/
    int find(vector<int>& nums1, int i, vector<int>& nums2, int j, int k) {
        //为了方便判断边界，假设nums1比nums2短
        if (nums1.size() - i > nums2.size() - j) return find(nums2, j, nums1, i, k);
        if (k == 1) {   //当k等于1, 说明需要找到第1小的数
            if (i >= nums1.size()) return nums2[j];  //边界情况，当k=1时，nums1不一定有元素，因为比nums2短
            else return min(nums1[i], nums2[j]);    //如果都有元素，取两个数组的最小值
        }
        if (i >= nums1.size()) return nums2[j + k - 1]; //当nums1没有元素，最小值在nums2中
        //由于nums1比较短，有可能加上k/2会超出长度，另外注意nums1.size()返回值类型不是int
        int si = min((int)nums1.size(), i + k / 2), sj = j + k - k / 2;
        if (nums1[si - 1] < nums2[sj - 1]) { //由于k是从1开始的，i和j是0开始，所以si和sj是k/2的后一个数
            return find(nums1, si, nums2, j, k - (si - i));
        }
        else 
            return find(nums1, i, nums2, sj, k - (sj - j));
    }
};