4. Median of Two Sorted Arrays 通俗易懂思路

There are two sorted arrays nums1 and nums2 of size m and n respectively.

Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).

从两个已经排序的数组中找出中位数，有时间复杂度限制。

---

用通俗易懂的说法说明一下思路：根据给出的复杂度推测应该要使用类似二分法的方法。

首先可以将问题一般化为：找到两个已排序数组中第k小的，数组总长度为奇数时找一次，为偶数时找两次即可得到答案。

怎么找呢？设两个数组长度分别为m, n, 比第k小的数大的数一定是m + n - k个，所以我们可以假设从两个数组中分别取出前 i 个和前 j 个，只要保证i + j <= k，那么比min(nums1[i - 1], nums2[j - 1])大的就会有至少有m + n - k + 1个（nums1和nums2未取出的后半部分和max(nums1[i - 1], nums2[j - 1])）（假设nums1[i - 1]和nums2[j - 1]不相等），那么就可以把较小的那个数之前的数全部删去，缩减数组规模，依此进行递归。

double getkth(int* nums1, int size1, int* nums2, int size2, int k){
    if(size1 > size2){
        return getkth(nums2, size2, nums1, size1, k);
    }
    if(size1 == 0){
        return nums2[k - 1];
    }
    if(k == 1){
        return nums1[0] < nums2[0] ? nums1[0] : nums2[0];
    }
    else{
        int i = size1 < k >> 1 ? size1 : k >> 1, j = k - i;
        if(nums1[i - 1] < nums2[j - 1])
            return getkth(nums1 + i, size1 - i, nums2, size2, k - i);
        else if(nums1[i - 1] > nums2[j - 1])
            return getkth(nums1, size1, nums2 + j, size2 - j, k - j);
        else
            return nums1[i - 1];
    }
}

double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) {
    int k1 = (nums1Size + nums2Size + 1) >> 1,
        k2 = (nums1Size + nums2Size + 2) >> 1;
    if(k1 == k2){
        return getkth(nums1, nums1Size, nums2, nums2Size, k1);
    }
    else{
        return (getkth(nums1, nums1Size, nums2, nums2Size, k1) + getkth(nums1, nums1Size, nums2, nums2Size, k2)) / 2;
    }    
}