Median of Two Sorted Arrays

题意：

给数组a[0],a[1]...a[m-1]

和数组b[0],b[1]...b[n-1]

两个数组均已排序

求中位数

难度：3

解法:

之前想了一个logm+logn的方法，写残了。看了别人的解法，理清思路

原问题引申为在两个排序数组中寻找第k小的数，以下为方便，设定k为偶数

比较A[k/2-1]与B[k/2-1]

如果A[k/2-1]<B[k/2-1],那么对于A[k/2-1],比它小的数最多只有k-2个，

也即，A[k/2-1]最好情况也是第k-1小的数，所以A[0]...A[k/2-1]都一定不是第k小的数，所以这段可以抛弃

所以转化为在A[k/2]...A[m-1]与B[0]....B[n-1]寻找第k/2小的数（因为第k小的数前面有k/2个数被抛弃了）

如果A[k/2-1]=B[k/2-1],那么第k小的数一定为两者中的较大值，对于A[k/2-1]与B[k/2-1]，一定有k-2个数比它们小，剩下的数全部比它们大，所以它们一定是第k-1小，第k小的数

class Solution
{
public:
    double findKth(int A[], int m, int B[], int n ,int k)
    {
        //比较A[ia],B[ib]
        if(m == 0)
        {
            return B[k-1];
        }
        if(n == 0)
        {
            return A[k-1];
        }
        if(k == 1)
        {
            return min(A[0],B[0]);
        }
        int ia=min(k/2-1,m-1);
        int ib=min(k/2-1,n-1);
        if(A[ia]>B[ib])
        {
            return findKth(A,m,B+ib+1,n-ib-1,k-ib-1);
        }
        else if(A[ia]<B[ib])
        {
            return findKth(A+ia+1,m-ia-1,B,n,k-ia-1);
        }
        else
        {
            if(ia+ib != k-2)
            {
                return findKth(A+ia+1,m-ia-1,B+ib+1,n-ib-1,k-ia-ib-2);
            }
            else
            {
                return max(A[ia],B[ib]);
            }
        }
    }
    double findMedianSortedArrays(int A[], int m, int B[], int n)
    {
        if((m+n)&1)
        {
            return findKth(A,m,B,n,(m+n+1)>>1);
        }
        else
        {
            return (findKth(A,m,B,n,(m+n)>>1)+findKth(A,m,B,n,(m+n+2)>>1))/2.0;
        }
    }
};