LeetCode 4. 寻找两个正序数组的中位数 【递归】

给定两个大小为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的中位数。

进阶：你能设计一个时间复杂度为 O(log (m+n)) 的算法解决此问题吗？

示例 1：

输入：nums1 = [1,3], nums2 = [2]
输出：2.00000
解释：合并数组 = [1,2,3] ，中位数 2

示例 2：

输入：nums1 = [1,2], nums2 = [3,4]
输出：2.50000
解释：合并数组 = [1,2,3,4] ，中位数 (2 + 3) / 2 = 2.5

示例 3：

输入：nums1 = [0,0], nums2 = [0,0]
输出：0.00000

示例 4：

输入：nums1 = [], nums2 = [1]
输出：1.00000

示例 5：

输入：nums1 = [2], nums2 = []
输出：2.00000

提示：

nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-106 <= nums1[i], nums2[i] <= 106

因为时间复杂的要求为log(n + m)，可以往二分的方向想。
题意可以转换为求第k小的元素，每次排除k >> 1 个元素，直至k == 1便可得出答案。

ps: & 的优先级比 == 低，记得加括号。在这里WA了一发。。

class Solution {
public:
    double getAns(vector<int>a, vector<int>b, int l1, int l2, int k) {
        int len1 = a.size() - l1, len2 = b.size() - l2;
        if(len1 > len2) return getAns(b, a, l2, l1, k);//令第一个数组总是长度最小的
        if(len1 == 0)   return b[l2 + k - 1];//若第一个数组剩余长度为零，返回答案
        if(k == 1)  return min(a[l1 + k - 1], b[l2 + k - 1]);//当前求的是最小的元素，直接返回最小值
        
        int x = min((int)a.size() - 1, l1 + k / 2 - 1);//a数组的对应下标
        int y = min((int)b.size() - 1, l2 + k / 2 - 1);
        
        if(a[x] < b[y])//排除a的前(x - l1 + 1)个元素
            return getAns(a, b, x + 1, l2, k - (x - l1 + 1));
        else
            return getAns(a, b, l1, y + 1, k - (y - l2 + 1));
    }
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int x = nums1.size(), y = nums2.size();
        int k = (x + y + 1) >> 1;
        double ans1 = getAns(nums1, nums2, 0, 0, k);//0 0 k分别表示两个数组的当前起始位置，当前求的是第k小的元素
        if((x + y) % 2 == 0) {//若总元素个数为偶数，要再求一次
            double ans2 = getAns(nums1, nums2, 0, 0, k + 1);
            // cout << ans1 << " " << ans2 << endl;
            return (ans1 + ans2) / 2;
        }
        return ans1;
    }
};