4. Median of Two Sorted Arrays

Problem:

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)).

Example 1:

nums1 = [1, 3]
nums2 = [2]

The median is 2.0

Example 2:

nums1 = [1, 2]
nums2 = [3, 4]

The median is (2 + 3)/2 = 2.5

题意是给出两个已经排好序的数组，然后找出两个数组组合起来后的中位数。初看问题，觉得很简单，可以用mergeSort排序然后直接找到第K项。这样的话复杂度是O(m+n)，但是题目要求算法复杂度大致为O(log(m+n))，所以这样做会超时。看到O(logn)级别的复杂度后，我觉得可以用类似二分查找的算法去寻找第k项。

假设m和n都大于k/2，然后判断nums1[k/2-1]和nums2[k/2-1]，假如是<的话，那么nums1的前k/2-1项就可以删掉，然后再判断剩下的nums1和nums2，并且k-k/2，一直到k==1或者nums1[k/2-1]==nums2[k/2-1]，可以停止，然后返回值。

 Code：(LeetCode运行33ms）

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int m = nums1.size();
        int n = nums2.size();
        int sum = m + n;
        if (sum % 2 == 1) {
            return (double) find_kth(nums1.begin(), m, nums2.begin(), n, sum / 2 + 1);
        } else {
            return (double) (find_kth(nums1.begin(), m, nums2.begin(), n, sum / 2) + find_kth(nums1.begin(), m, nums2.begin(), n, sum / 2 + 1)) / 2.0;
        }
    }
    int find_kth(vector<int>::const_iterator A, int m, vector<int>::const_iterator B, int n, int k) {
        if (m > n) {
            return find_kth(B, n, A, m, k);
        }
        if (m == 0) {
            return *(B + k - 1);
        }
        if (k == 1) {
            return min(*A, *B);
        }
        
        //划分k到nums1和nums2上。
        int i = min(k/2, m), j = k - i;
        if (*(A + i - 1) < *(B + j - 1)) {
            return find_kth(A + i, m - i, B, n, k - i);
        } else if (*(A + i - 1) > *(B + j - 1)) {
            return find_kth(A, m, B + j, n - j, k - j);
        } else {
            return *(A + i - 1);
        }
    }
};