二分寻找两个正序数组的中位数

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

算法的时间复杂度应该为 O(log (m+n)) 。

题目来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/median-of-two-sorted-arrays
 

 

 

class Solution {

public:

    double rekv(vector<int> &a,int l1,int r1,vector<int> &b,int l2,int r2,int k){

    //k表示要返回的下标

    if(l1>r1){

        //第一个数组没有

        return b[l2+k]; 

    }

    else if(l2>r2){

        return a[l1+k];

    }

    else if(k==0){

        //返回下标为0的数，只要找出数组中较小的数即可

        return a[l1]<b[l2] ? a[l1]:b[l2];

    }

    else{

        int d;

        if(k%2==0){

            //返回的下标是偶数

            d=k/2-1;  //下表d+l1就是要比较的数

        }

        else{

            d=k/2;

        } 

        if(d>(r1-l1)||d>(r2-l2)){

            if(d>r1-l1){

                //超过第一个数组

                if(a[r1]<=b[l2+d]){

                    //把第一个数组排除

                    return b[k-(r1-l1+1)]; 

                }

                else{

                    //把第一个数组留下 

                    return rekv(a,l1,r1,b,l2+d+1,r2,k-(d+1));

                }

            }

            else{

                //超过第二个数组 

                if(a[l1+d]>=b[r2]){

                    //把第二个数组排除

                    return a[k-(r2-l2+1)]; 

                }

                else{

                    return rekv(a,l1+d+1,r1,b,l2,r2,k-(d+1));

                }

            } 

        }

        else{

            //没有超过数组

            if(a[l1+d]<=b[l2+d]){

                //排除第一个数组的前面一部分

                return rekv(a,l1+d+1,r1,b,l2,r2,k-(d+1));

            } 

            else{

                //排除第二个数组的前一部分 

                return rekv(a,l1,r1,b,l2+d+1,r2,k-(d+1));

            }

        }

    }

    }

    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {

        int m,n,k;

        //k表示要返回的下标所在的数

        double ans=0;

        m=nums1.size();n=nums2.size();

        if((m+n)%2==0){

            ans=(rekv(nums1,0,m-1,nums2,0,n-1,(m+n)/2)+rekv(nums1,0,m-1,nums2,0,n-1,(m+n)/2-1))/2;

        }

        else{

            ans=rekv(nums1,0,m-1,nums2,0,n-1,(m+n)/2);

        }

        return ans;

    }

};