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

大致题意就是：

对于两个排好序的数组nums1和nums2，size分别为m和n，找到两个数组合并后的中位数。要求时间复杂度在O(log(m + n))

其中要注意的是：如果数组size和是偶数的话则中位数是中间两个数的平均值，且函数返回值为double。

这个题见过两个思路，一个是经典的findKth，九章上有解答，复杂度是O(log(range)∗(log(n)+log(m)))：

https://www.jiuzhang.com/solution/median-of-two-sorted-arrays/

下面这个思路是top solution里面的一个方法，本质也是二分法。不失一般性，先假设数组nums1长度小于等于nums2，且总长度为奇数。如果找到了中位数nums1[i - 1]或nums2[j - 1]，根据定义可知nums1和nums2一定会分别被分为如下情况：

nums1: 0, 1, 2, ..., i - 1，          i, ...., n - 1

                 左侧                       右侧  

nums2: 0, 1, 2, ..., j - 1，         j, ..., m - 1

使得nums1[i - 1] <= nums2[j] && nums2[j - 1] <= nums1[i]， 且i + j == (m + n + 1) / 2（不考虑越界的话）

有了这三个条件，我们就可以对nums1进行二分查找来找到一个满足条件的i，然后左边最大的数就是中位数。其中比较tricky的地方就是边界判断，一个是i或者j取到了0或者取到了n或者m，另一个是如果总长度是偶数的话得把右边最小的数加上求平均。

c++代码：

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int n = nums1.size(), m = nums2.size();
        if (n == 0 && m == 0) return 0.0;
        if (n > m) return findMedianSortedArrays(nums2, nums1);
        int left = 0, right = n;
        int i = 0, j = 0;
        while (left <= right) {
            i = left + (right - left) / 2;
            j = (n + m + 1) /2 - i;
            if (i < n && nums1[i] < nums2[j - 1]) {
                left = i + 1;
            } else if (i > 0 && nums2[j] < nums1[i - 1]) {
                right = i - 1;
            } else {
                break;
            }
        }
        double leftMax = 0;
        if (i == 0) {
            leftMax = nums2[j - 1];
        } else if (j == 0) {
            leftMax = nums1[i - 1];
        } else {
            leftMax = max(nums1[i - 1], nums2[j - 1]);
        }
        if (1 & (n + m)) return leftMax;
        double rightMin = 0;
        if (i == n) {
            rightMin = nums2[j];
        } else if (j == m) {
            rightMin = nums1[i];
        } else {
            rightMin = min(nums1[i], nums2[j]);
        }
        return (rightMin + leftMax) / 2;
    }
};

原题链接：https://leetcode.com/problems/median-of-two-sorted-arrays/description/

从上一篇之后偷懒了好久，一直在刷题。。终于可以休息一下，稍微写点东西