LeetCode 4. Median of Two Sorted Arrays

这道题很经典，它归类在分治算法，感觉和求一堆数的第k大差不多，不过这里是有两个已经排好序的数组，所以比较麻烦一些。整体的思路如下：
1、要求出两个数列的中间数，可以转换成求第k小（大）的数；
2、要求第k小的数，可以分为将第一个数列的k/2个数和第二个数列的k/2个数进行对比，小的那一个数列的左边k/2个数就可以丢弃了，如此递归下去，递归的终止条件为：（1）如果有一个数列的长度为0，则取出另一个数列的第k大；（2）如果k=1，则直接比较两个数列的第一个数然后取出较小的数即可；（3）这里可以假设第一个数列总是比第二个数列要短，方便写代码。
代码如下：

int min(int a, int b){
    return a<b?a:b;
}

double FindKthSmallest(int* num1, int m, int* num2, int n, int k)
{
    if(m > n) //保证num1总是比num2短
        return FindKthSmallest(num2, n, num1, m, k);
    if(m == 0)
    {
        return num2[k-1];
    } 
    if(k == 1)//这里防止了下面的pa-1小于0
    {
        return min(num1[0],num2[0]);
    }
    int pa = min(m, k/2), pb = k-pa;//pb不能取min(n,k/2),否则过不了[1,2]和[1,2]这种样例

    if(num1[pa-1] < num2[pb-1])
        return FindKthSmallest(num1+pa, m-pa, num2, n, k-pa);
    else if(num1[pa-1] > num2[pb-1])
        return FindKthSmallest(num1, m, num2+pb, n-pb, k-pb);
    else
        return num1[pa-1];

}

double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size)
{
    int k = (nums1Size+nums2Size)/2;
    if((nums1Size+nums2Size)%2 == 0)
    {   
        int a = FindKthSmallest(nums1, nums1Size, nums2, nums2Size, k);
        int b = FindKthSmallest(nums1, nums1Size, nums2, nums2Size, k+1);
        return (a+b)/2.0;
    }
    else 
        return FindKthSmallest(nums1, nums1Size, nums2, nums2Size, k+1);
}