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

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

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int sum = nums1.size() + nums2.size();
        if(sum % 2 == 0){//偶数的情况：返回中间两数平均值
            int left = find(nums1,0,nums2,0,sum/2);
            int right = find(nums1,0,nums2,0,sum/2 + 1);
            return (left+right)/2.0;
        }
        else return find(nums1,0,nums2,0,sum/2 + 1);//奇数情况，返回中间的数
    }

     int find(vector<int> &nums1,int i,vector<int> &nums2,int j,int k){//找第k个数
        if(nums1.size()-i > nums2.size()-j) return find(nums2,j,nums1,i,k);//假定第一个数组比较短，若长则交换
        if(nums1.size() == i) return nums2[j+k-1]; //第一个数组为空，返回第二个数组第k个
        if(k == 1){//只有1个数
            if(nums1.size()==i) return nums2[j];//第一个数组为空 返回第二个数组
            else return min(nums2[j],nums1[i]);//返回两个的最小值
        }
        int si=min((int)i+k/2,int(nums1.size())),sj=j+k-k/2;
        if(nums1[si-1]>nums2[sj-1]){
            return find(nums1,i,nums2,sj,k-(sj-j));// 减去nums2前K/2的数
        }
        else{
            return find(nums1,si,nums2,j,k-(si-i));//减去nums1的前K/2的数
        }
    }
};