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

• 维护两个数组的当前索引，逐步向前移动，直到移动到中位数所在的索引。

• 代码

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int length = nums1.size() + nums2.size();
        int index_1 = length % 2 != 0 ? length / 2 : length / 2 - 1;
        int flag = 0;
        int index_nums1 = -1;
        int index_nums2 = -1;
        double num_sum = 0.0;
        int res = 0;

        for (int i = 0; i < length; i++) {
            // 情况 1: nums1 已经遍历完
            if (index_nums1 + 1 >= nums1.size()) {
                index_nums2++;
                flag = 1;
            } 
            // 情况 2: nums2 已经遍历完
            else if (index_nums2 + 1 >= nums2.size()) {
                index_nums1++;
                flag = 0;
            } 
            // 情况 3: 两个数组都没遍历完，比较当前元素大小
            else {
                if (nums1[index_nums1 + 1] >= nums2[index_nums2 + 1]) {
                    index_nums2++;
                    flag = 1;
                } else {
                    index_nums1++;
                    flag = 0;
                }
            }

            // 处理中位数的第一个数
            if (i == index_1) {
                if (length % 2 != 0) {
                    res = flag == 0 ? nums1[index_nums1] : nums2[index_nums2];
                    return res;
                } else {
                    res = flag == 0 ? nums1[index_nums1] : nums2[index_nums2];
                    num_sum = res / 2.0;
                }
            }
            // 处理中位数的第二个数
            if (i == index_1 + 1) {
                res = flag == 0 ? nums1[index_nums1] : nums2[index_nums2];
                num_sum += res / 2.0;
                return num_sum;
            }
        }
        return 0.0;
    }
};