Medium of Two Sorted Arrays

//递归的二分查找
class Solution 
{
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) 
    {
        int m = nums1.size();
        int n = nums2.size();
        if ((m+n)%2 == 0)
            return (findKthNum(nums1, nums2, (m + n) / 2) + findKthNum(nums1, nums2, (m + n) / 2 + 1))/2;
        else
            return findKthNum(nums1, nums2, (m + n) / 2 + 1);
    }
    int min(int a, int b)
    {
        if (a < b)
            return a;
        else
            return b;
    }
 
    double findKthNum(vector<int>& nums1, vector<int>& nums2, int k)
    {
        if (nums1.size() > nums2.size())
        {
            return findKthNum(nums2, nums1, k);  //注意nums1的size一定要比nums2小，不然会发生溢出错误
        }
        //当m或者n为0时 //always assume that m is equal or smaller than n
        if (nums1.size() == 0)
        {
                return nums2[k - 1];
        }
        if (k == 1)//当k为1时 不用算了 直接搞
        {
            return min(nums1[0], nums2[0]);
        }
        int offset_1 = min(k / 2, nums1.size());
        int offset_2 = k - offset_1;        
        vector<int> new_nums1, new_nums2;
        int i = 0;
        if (nums1[offset_1-1] == nums2[offset_2-1])
        {
            return nums1[offset_1-1];
        }
        else if (nums1[offset_1-1] < nums2[offset_2-1])
        {
            for (vector<int>::iterator it = nums1.begin(); it != nums1.end(); it++)
            {
                if (i >= offset_1)
                {
                    new_nums1.push_back(*it);
                }
                i++;
            }
            return findKthNum(new_nums1, nums2, k - offset_1);
        }
        else
        {
            for (vector<int>::iterator it = nums2.begin(); it != nums2.end(); it++)
            { 
                if (i >= offset_2)
                {
                    new_nums2.push_back(*it);
                }
                i++;
            }
            return findKthNum(nums1, new_nums2, k - offset_2);
        }
    }
};

核心思想是将原问题转变成一个寻找第k小数的问题（假设两个原序列升序排列），这样中位数实际上是第(m+n)/2小的数。所以只要解决了第k小数的问题，原问题也得以解决。首先假设数组A和B的元素个数都大于k/2，我们比较A[k/2-1]和B[k/2-1]两个元素，这两个元素分别表示A的第k/2小的元素和B的第k/2小的元素。这两个元素比较共有三种情况：>、<和=。如果A[k/2-1]<B[k/2-1]，这表示A[0]到A[k/2-1]的元素都在A和B合并之后的前k小的元素中。换句话说，A[k/2-1]不可能大于两数组合并之后的第k小值，所以我们可以将其抛弃。