按照顺序排列的两个数组,数组A:[a1,a2,…,an],数组B:[b1,b2,…,bn],求合并后的数组C:[c1,c2,…,c2n]的中位数
1、求A、B两数组各自的中位数m1,m2;
2、比较m1,m2,:
若m1 = m2,合并后中位数仍等于m1;
若m1 < m2,舍弃数组A左半部分和数组B右半部分,要求舍弃的元素数目一致
若m1 > m2,舍弃数组A右半部分和数组B左半部分,要求舍弃的元素数目一致
3、重复步骤2直至A、B两数组都只剩一个元素
C++实现:
int M_search(int A[], int B[], int n) {
int s1 = 0, d1 = n - 1, m1, s2 = 0, d2 = n - 1, m2;
while (s1 != d1 || s2 != d2) //重复该过程直至两数列中仅剩一个元素
{
m1 = (s1 + d1) / 2;
m2 = (s2 + d2) / 2; //表示两个数列的中位数
if (A[m1]==B[m2]) //步骤1:若两个数列中位数ab相等,则a即为所求
{
return A[m1];
}
if (A[m1] < B[m2]) //步骤2:若两个数列中位数a<b,则去掉数列A左半部分和数列B右半部分
{
if ((s1+d1)%2==0) //量数列剩余项数为奇数,等价于(d1-s1)%2==0,(2,8)共7个数,保留中位数a点
{
s1 = m1;
d2 = m2;
}
else //量数列剩余项数为偶数,等价于(d1-s1)%2==1,(2,7)共6个数,去除前一半时不保留中位数a点
{
s1 = m1 + 1;
d2 = m2;
}
}
else
{
if ((s1 + d1) % 2 == 0) //量数列剩余项数为奇数,等价于(d1-s1)%2==0,(2,8)共7个数,去除前一半时不保留中位数a点
{
s2 = m2;
d1 = m1;
}
else //量数列剩余项数为偶数,等价于(d1-s1)%2==1,(2,7)共6个数,保留中位数a点
{
s2 = m2 + 1;
d1 = m1;
}
}
}
return A[s1] < B[s2] ? A[s1] : B[s2];
}