问题描述:
给定两个有序数组arr1和arr2,两个数组长度都为N,求两个数组中所有数的上中位数。
例如:
arr1 = {1,2,3,4};
arr2 = {3,4,5,6};
一共8个数则上中位数是第4个数,所以返回3。
arr1 = {0,1,2};
arr2 = {3,4,5};
一共6个数则上中位数是第3个数,所以返回2。
要求:时间复杂度O(logN)
最优解代码:
class Solution {
public:
int getupmedian(vector<int> &arr1, int arr1b,int arr1e,vector<int> &arr2,int arr2b,int arr2e){
int mid1=(arr1e+arr1b)/2;
int mid2=(arr2e+arr2b)/2;
int offest=((arr1e-arr1b+1)&1)^1;
if(arr1e<arr1b)
return -1;
if(arr1e==arr1b)
return arr1[arr1b]>arr2[arr2b]?arr2[arr2b]:arr1[arr1b];
if(arr1[mid1]==arr2[mid2])
return arr1[mid1];
else if(arr1[mid1]>arr2[mid2])
return getupmedian(arr1, arr1b,mid1,arr2,mid2+offest,arr2e);
else return getupmedian(arr1, mid1+offest,arr1e,arr2,arr2b,mid2);
}
int getUpMedian(vector<int> arr1, vector<int> arr2) {
int len=arr1.size();
return getupmedian(arr1,0,len-1,arr2,0,len-1);
}
};