在两个长度相等的排序数组中找到上中位数
程序员文章站
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]);
}
};