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

力扣 num4. 寻找两个正序数组的中位数

链接: 题目.
用递归的方法，时间复杂度O(log(m+n))

思路:

把两个数组想象成合并的样子，但是每个数组我们都单独去处理。
首先计算出k=(n+m),然后对比两个数组中的num1[k/2]/,num2[k/2],
我们去对比两个数字的大小，有三种情况：

1:n1 > n2，说明在num2的前k/2个数中，没有一个数是大于第k小的数的，这样我们就可以去删除num2中的这钱k/2个数

2： n1 == n2，说明num[k/2]或num2[k/2]恰好是第k个小的数。为什么说是或呢，因为我们在第一次划分的时候，在num1中的num[k/2]前面有k/2-1个元素，而在num2中也是k/2-1个元素，这样第k个元素是num1[k/2]或num2[k/2]中较大的那一个,因为2*(k/2-1)=k-2，第k个元素是k-2+2，就是num1[k/2],num2[k/2]中较大的那一个。
3: 与第一种情况相反就不再赘述。

代码

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int tot =nums1.size()+nums2.size();
        //如果两个数组的元素都是偶数个
        if(tot%2==0){
            int left=find(nums1,0,nums2,0,tot/2);
            //把两个数组放一起 找到第k/2和第k/2+1个数
            //这里的 k/2是指合并的两个数组的第k/2个小的数
            int right=find(nums1,0,nums2,0,tot/2+1);
            return (left+right)/2.0;
        }
        else return find(nums1,0,nums2,0,tot/2+1);
    }
    //find函数是 第一个数组 从i位置开始找 第二个数组从j位置开始找 找到第k个小的数
    int find(vector<int>&num1,int i,vector<int>&num2,int j,int k){
        //这里默认num1是小的数组

        if(num1.size()-i>num2.size()-j) return find(num2,j,num1,i,k);
        //处理边界
        //当两个数组只有一个数字或者只有一个有一个数字的时候
        if(k==1){
            //第一个数组没有元素了 因为默认第一个数组最小而k==1 说明找第一个小的数 
            //那么这个数一定在第二个数组里面
            if(num1.size()==i) return num2[j];//num1空的时候 就是num2中的数字
            //否则k==1 的情况一定是两个数组开头的最小的数了 这个没有疑问
            else return min(num1[i],num2[j]);
        }
        //当num1空的时候
        if(num1.size()==i) return num2[j+k-1];//当第一个数组空的时候 就是从第二个数组
                                             //里面开始查第k个数

        int si=min(i+k/2,(int)num1.size()),sj=j+k-k/2;
        if(num1[si-1]>num2[sj-1]) return find(num1,i,num2,sj,k-(sj-j));
        else return find(num1,si,num2,j,k-(si-i));
    }
};