LeetCode 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
思路
med = m + n，若k为偶则，取第med/2和med/2+1的数的平均值。若k为奇，则取第med/2+1。
写一个find 函数int find(vector nums1, int i, vector nums2, int j, int k
（递归，设定短数组在前，长数组在后，i、j分别为nums1，nums2的起始位置，该函数可找出总排列中的第k个元素）
1.只有一个元素
2.第一个数组是空的：找出第二个数组第k个元素
3.找出第一个数组的第k/2个元素si（同时应考虑i+k/2越界情况），第二个数组的k-k/2个元素sj，若第一个元素小于第二个元素，则丢弃第一个数组的前k/2元素，nums1起始位置改为si，k减去k/2，问题变为找出总排列的第k-k/2个元素，递归。
若第一个元素大于第二个元素，则丢弃第二个数组的前k-k/2元素，nums2起始位置改为sj，k减去k-k/2，问题变为找出总排列的k/2个元素，递归。
y总代码

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int m = nums1.size();
        int n = nums2.size();
        int med = m + n;
        if(med % 2 == 0){
            int left = find(nums1, 0, nums2, 0, med / 2);
            int right = find(nums1, 0, nums2, 0, med / 2 + 1);
            return (left + right)/2.0;
        }else{
            return find(nums1, 0, nums2, 0, med/2 + 1);
        }
        
    }

    int find(vector<int> nums1, int i, vector<int> nums2, int j, int k){
        //假定nums1.size()<nums2.size()
        if(nums1.size() - i > nums2.size() - j) return find(nums2, j, nums1, i, k);
        if(k == 1){//只有一个，取nums1,nums2第一个最小值
            if(nums1.size() == i) return nums2[j];//第一个数组是空的
            else return min(nums1[i],nums2[j]);
        }
        if(nums1.size() == i) return nums2[j + k-1];
        int si = min((int)nums1.size(), i + k/2) , sj = j + k - k/2;
        if(nums1[si - 1] < nums2[sj - 1]) return find(nums1, si, nums2, j, k - (si - i));
        else return find(nums1, i, nums2, sj, k - (sj - j));
    }
};