在两个长度相等的排序数组中找到上中位数

给定两个有序数组arr1和arr2,已知两个数组的长度都为N,求两个数组中所有数的上中位数。
上中位数:假设递增序列长度为n,若n为奇数,则上中位数为第n/2+1个数;否则为第n/2个数
[要求]
时间复杂度为O(logN)O(logN),额外空间复杂度为O(1)O(1)
示例1

[1,2,3,4],[3,4,5,6]

输出

3

说明
总共有8个数,上中位数是第4小的数,所以返回3。
示例2
输入

[0,1,2],[3,4,5]

输出

2

说明
总共有6个数,那么上中位数是第3小的数,所以返回2

class Solution {
public:
    /**
     * find median in two sorted array
     * @param arr1 int整型vector the array1
     * @param arr2 int整型vector the array2
     * @return int整型
     */
    int findMedianinTwoSortedAray(vector<int>& arr1, vector<int>& arr2) {
        int l1 = 0, r1 = arr1.size() - 1, l2 = 0, r2 = arr2.size() - 1;
        
        while(l1 < r1){
            int mid1 = (l1 + r1) >> 1, mid2 = (l2 + r2) >> 1;
            //判断子区间的个数是奇数还是偶数 0代表是奇数(奇数中间的数字不能删除) 1代表是偶数
            int even = (r1 - l1) % 2;
            if(arr1[mid1] == arr2[mid2]){
                return arr1[mid1];
            } else if (arr1[mid1] > arr2[mid2]){
                //证明arr1前面的子数组没用 arr2后面的子数组没用
//                 r1 = mid1 + even; //如果是奇数就是啥都不做 
                /*
                3456 1234 得取34所有 l2 应该向后面取一位
                */
                r1 = mid1;
                l2 = mid2 + even;
            } else {
                l1 = mid1 + even; //如果是偶数 就是它的后一位
                r2 = mid2; 
            }
        }
        //如果俩个数组剩下最后一个数 且不相等 返回其中较小的那个即可
        return min(arr1[l1],arr2[l2]);
    }
};

猜你喜欢

转载自blog.csdn.net/weixin_43599304/article/details/115378605