Leetcode 4. 寻找两个正序数组的中位数

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

进阶：你能设计一个时间复杂度为 O(log (m+n)) 的算法解决此问题吗？
可以推广为：给定两个有序数组，在log时间复杂度内找出第k大的数。

数组A=[A0,A1,…,AM-1],B=[B0,B1,…,BN-1]。
假设找到了前k大的数集合C，则C=[A0,A1,…,Ai,B0,B1,…Bj]（即必为A,B前缀的并）。
且具有以下性质：Ai<=Bj+1,Bj<=Ai+1，i+1+j+1=k。
第k大的数=max{Ai,Aj}。
也可以写成i=k-2-j，即若枚举i，可直接求出j。

由于A,B均有单调性，可以把上述问题转换为：
找到最大的i，满足Ai<=Bj+1。那么第k大的数即为max{Ai,Aj}。
这个二分即可。
注意边界条件，把<0设为-INF，>0设为INF即可。（边界害得我WA了好多遍）

---

有了第k大的做法，中位数就很好求了。
如果是奇数，则求第（len+1）大的数。
如果是偶数，则答案为(find(len/2)+find(len/2+1))/2。

---

代码：

#define INF 0xfffffff

class Solution {
private:
    int ffind_k(vector<int>& nums1, vector<int>& nums2,int k)
    {
        int m = nums1.size(),n = nums2.size();
        int l=-1,r=m-1,ans;
        if(m==0) return nums2[k-1];
        if(n==0) return nums1[k-1];
        while(l<=r)
        {
            int mid=(l+r)/2;
            int a1 = mid<0?-INF : nums1[mid];
            int b1 = k-2-mid<0?-INF: (k-2-mid>=n?INF:nums2[k-2-mid]) ;
            int b2 = k-2-mid+1<0?-INF: (k-2-mid+1>=n?INF:nums2[k-2-mid+1]) ;
            if(a1<=b2)
            {
                ans = max(a1,b1);
                l=mid+1;
            }
            else r=mid-1;
        }
        return ans;
    }
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int len = nums1.size() + nums2.size();
        if(len%2==1)
            return ffind_k(nums1,nums2,(len+1)/2)*1.0;
        else
            return (ffind_k(nums1,nums2,len/2)+ffind_k(nums1,nums2,len/2+1))*0.5;
    }
};