力扣 4. 寻找两个有序数组的中位数 递归+二分

https://leetcode-cn.com/problems/median-of-two-sorted-arrays/

思路：这个复杂度限定还是挺有意思的。假设$getkth(k)$可以求出数组$a$和数组$b$后并之后的第$k$个数，设数组$a、b$的长度分别为$n_1、n_2$，设$k_1=(n_1+n_2+1)/2,k_2=(n_1+n_2+2)/2$，那么数组$a、b$的中位数就等于$(getkth(k_1)+getkth(k_2))/2.0$。这个比较简单，大家举几个例子就能推出来。现在我们考虑怎么实现这个$getkth$函数。我们可以设$kk=k/2$，那么通过比较$a[kk]$和$b[kk]$的大小，就可以筛去一部分不合题意的数字。假设$a[kk]>=b[kk]$，$a$数组中$<=a[kk]$的数共有$kk-1$个，$b$数组中$<=b[kk]$的数共有$kk-1$个，就算$b[kk]>=a[kk-1]$，那么小于$b[kk]$的数最多只有$2\ast kk-2$个，也就是说$b[kk]$最大也不过是第$k-1$大的数，由此我们可以看出$b[0\dots kk]$均不是符合题意的解，因此可以把这部分数筛去，问题转换成求两个新数组的第$k-(kk+1)$大的数。那么递归进行这个过程，就可以求得解。为了方便起见，我们在函数中新增$4$个参数$be{g}_1,en{d}_1,be{g}_2,en{d}_2$用来限制$a、b$的范围，这样可以使得筛去操作容易实现。递归的出口就是$be{g}_1=en{d}_1$或$be{g}_2=en{d}_2$或$k=1$。同时考虑到某个数组的实际个数可能小于$kk$，在限定范围的时候需要注意。

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int n1=nums1.size();
        int n2=nums2.size();
        int k1=(n1+n2+1)>>1;
        int k2=(n1+n2+2)>>1;
        return (getkth(nums1,0,n1,nums2,0,n2,k1)+getkth(nums1,0,n1,nums2,0,n2,k2))/2.0;
    }

    int getkth(vector<int>& nums1,int beg1,int end1,vector<int>& nums2,int beg2,int end2,int k){
        int len1=end1-beg1;
        int len2=end2-beg2;
        if(len1==0)
            return nums2[beg2+k-1];
        if(len2==0)
            return nums1[beg1+k-1];
        if(k==1)
            return min(nums1[beg1],nums2[beg2]);
        int id1=beg1+min(len1,k>>1)-1;
        int id2=beg2+min(len2,k>>1)-1;
        if(nums1[id1]>=nums2[id2])
            return getkth(nums1,beg1,end1,nums2,id2+1,end2,k-(id2-beg2+1));
        else
            return getkth(nums1,id1+1,end1,nums2,beg2,end2,k-(id1-beg1+1));
    }
};