【LeetCode 4.】Median of Two Sorted Arrays

题目大意：

给两个有序数组，求出合并以后的中位数。
要求时间复杂度：O(m+n)

思路：

二分。

代码

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        if (nums1.size() > nums2.size()) {
            swap(nums1, nums2);
        }
        
        int n1 = nums1.size();
        int n2 = nums2.size();
        int l = 0, r = n1;
        while (true) {
            int i = (l + r) / 2;
            int j = (n1 + n2 + 1) / 2 - i;
            if ((j == 0 || i == n1 || nums1[i] >= nums2[j-1]) && (i == 0 || j == n2 || nums2[j] >= nums1[i-1])) {
                int left_max = 0;
                if (i == 0) left_max = nums2[j-1];
                else if (j == 0) left_max = nums1[i-1];
                else left_max = max(nums1[i-1], nums2[j-1]);
                
                if ((n1 + n2) % 2 != 0) return left_max;
                
                int right_min = 0;
                if (i == n1) right_min = nums2[j];
                else if (j == n2) right_min = nums1[i];
                else right_min = min(nums1[i], nums2[j]);
                return (left_max + right_min) * 1.0 / 2;
            }
            
            else if (j < n2 && i > 0 && nums2[j] < nums1[i-1]) {
                r = i - 1;
            }
            else if (i < n1 && j > 0 && nums1[i] < nums2[j-1]) {
                l = i + 1;
            }
        }
        
        return -1;
    }
};

参考链接：https://juejin.im/post/5c8c9ec16fb9a049af6e2a0d