寻找两个有序数组的中位数:题目描述
思路
如图B数组5个元素,A数组7个元素
假如,A,B在合并排序的时候,b1,b2在前6个元素里面,则A的a1,a2,a3,a4一定在前6个元素
我们在A,B的每一个位置都预留一个位置用来分割,如下
我们看见c1的位置是B被划分到前面长度的2倍
c2的位置是A被划分到前面的元素个数的2倍
则c1 + c2 = 原来两个元素的长度
即n = B.size(),m = A.size()
则c1 + c2 = m + n
对于c1前面的数,可以表示为(c1 - 1)/2 或者(c1 - 2)/2都可以,对于c1后面的数,可以表示为c1/2 或者(c1 + 1)/2都可以
然后利用c1前面的数一定小于c2后面的数,c2前面的数一定小于c1后面的数来调整c1和c2的位置。
以上情况是偶数个的情况,下面我们看看奇数个的情况
如下c2指向的是某个数字,我们可以将c2前面的数表示为(c2 - 1)/2,c2后面的数表示为c2/2。其实他们都表示a4。
如果我们只看第二个数组
那么这个时候我们的中位数是a4。可以表示为(a4 + a4)/2。相当于强行把奇数个数变成了偶数个数。
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int n = nums1.size();
int m = nums2.size();
if(n > m) return findMedianSortedArrays(nums2,nums1);
int lo = 0,hi = 2 * n,c1,c2,l1,r1,l2,r2;
while(lo <= hi){
c1 = (lo + hi) / 2;
c2 = m + n - c1;
l1 = (c1 == 0)?INT_MIN:nums1[(c1 - 1)/2];
r1 = (c1 == 2 * n) ? INT_MAX:nums1[c1/2];
l2 = (c2 == 0)?INT_MIN:nums2[(c2 - 1)/2];
r2 = (c2 == 2 * m) ? INT_MAX:nums2[c2/2];
if(l1 > r2)
hi = c1 - 1;
else if(l2 > r1)
lo = c1 + 1;
else break;
}
return (max(l1,l2) + min(r1,r2)) * 1.0 / 2;
}
};