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

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

---

题目：

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

算法的时间复杂度应该为 O(log (m+n)) 。

示例：

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

解题思路：
 定义数组总长度int tot = nums1.size() + nums2.size()，判断数组为奇数还是偶数，若为奇数，则找到第tot / 2 + 1大的数返回；若为偶数，则找到第tot / 2大和第tot / 2 + 1大的数返回(left + right) / 2.0。

 然后定义查找算法（递归）。假设nums1数组短，若它较长，则交换。然后判断k的大小，若k==1代表两数组最多各只剩一个数，先判断短的数组nums1是否为空，若为空的话直接返回nums2的数组，否则返回两个数组的最小值。若nums1递归到空时，则直接返回nums2的第k个数。前面的操作都在定义终止条件，接着进行递归判断，定义si和sj为两个数组第k/2个元素的下一个位置。然后比较两数组第k/2元素值得大小，若nums1大，则nums2中前k/2个元素一定小于第k个数，可以舍去，否则舍去num1中前k/2个元素。以此递归，方可解题。

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int tot = nums1.size() + nums2.size();
        if (tot % 2 == 0) {
            int left = find(nums1, 0, nums2, 0, tot / 2);
            int right = find(nums1, 0, nums2, 0, tot / 2 + 1);
            return (left + right) / 2.0;
        } else {
            return find(nums1, 0, nums2, 0, tot / 2 + 1);
        }
    }
    int find(vector<int>& nums1, int i, vector<int>& nums2, int j, int k){
        // 假设nums1短，若nums1长的话，那么交换一下，先算nums2
        if (nums1.size() - i > nums2.size() - j) return find(nums2, j, nums1, i, k);
        // 当只剩一个数时，代表找到了最终的两个数
        if (k == 1) {
            // 若nums1为空的话，则返回nums2的那个数
            if (nums1.size() == i) return nums2[j];
            // 否则返回两者的最小值
            else return min(nums1[i], nums2[j]);
        } 
        // 若nums1为空了，返回nums2的第k个数
        if (nums1.size() == i) return nums2[j + k - 1];
        // 两个数组的第k/2个元素的下一个元素的位置
        int si = min((int) nums1.size(), i + k / 2), sj = j + k - k / 2;
        // 若nums1的第k/2个元素值大于nums2的第k/2的元素值，那么nums2的前半段无用
        if (nums1[si - 1] > nums2[sj - 1]) return find(nums1, i, nums2, sj, k - (sj - j));
        else return find(nums1, si, nums2, j, k - (si - i));
    }
};