leetcode:寻找两个有序数组的中位数

寻找两个有序数组的中位数：题目描述

思路

如图B数组5个元素，A数组7个元素

假如，A,B在合并排序的时候，b1,b2在前6个元素里面，则A的a1,a2,a3,a4一定在前6个元素

我们在A,B的每一个位置都预留一个位置用来分割，如下

我们看见c1的位置是B被划分到前面长度的2倍
c2的位置是A被划分到前面的元素个数的2倍
则c1 + c2 = 原来两个元素的长度
即n = B.size(),m = A.size()
则c1 + c2 = m + n

对于c1前面的数，可以表示为(c1 - 1)/2 或者(c1 - 2)/2都可以，对于c1后面的数，可以表示为c1/2 或者(c1 + 1)/2都可以
然后利用c1前面的数一定小于c2后面的数，c2前面的数一定小于c1后面的数来调整c1和c2的位置。

以上情况是偶数个的情况，下面我们看看奇数个的情况
如下c2指向的是某个数字，我们可以将c2前面的数表示为(c2 - 1)/2，c2后面的数表示为c2/2。其实他们都表示a4。

如果我们只看第二个数组

那么这个时候我们的中位数是a4。可以表示为(a4 + a4)/2。相当于强行把奇数个数变成了偶数个数。

    class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int n = nums1.size();
        int m = nums2.size();
        if(n > m) return findMedianSortedArrays(nums2,nums1);
        int lo = 0,hi = 2 * n,c1,c2,l1,r1,l2,r2;
        while(lo <= hi){
            c1 = (lo + hi) / 2;
            c2 = m + n - c1;
            l1 = (c1 == 0)?INT_MIN:nums1[(c1 - 1)/2];
            r1 = (c1 == 2 * n) ? INT_MAX:nums1[c1/2];
            
            l2 = (c2 == 0)?INT_MIN:nums2[(c2 - 1)/2];
            r2 = (c2 == 2 * m) ? INT_MAX:nums2[c2/2];
            
            if(l1 > r2)
                hi = c1 - 1;
            else if(l2 > r1)
                lo = c1 + 1;
            else break;
            
        }
        return (max(l1,l2) + min(r1,r2)) * 1.0 / 2;
        
    }
};