欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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

程序员文章站 2024-03-15 22:01:06
...

给定两个有序数组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]);
    }
};