题目:假设两个等长升序序列A、B; 设计一个时间复杂度尽可能高效、空间复杂度为O(1)的算法,找出两个序列A和B的中间数。
设计思路:
1、设m1、m2分别为序列A、B中位数,判断m1、m2的关系:
- 如果 m1=m2,说明找到两个序列共同的中位数,即题目所求中位数,算法结束,
- 如果 m1<m2,则需要舍去序列A中较小的一半以及序列B中较大的一半,要求舍去长度相等
- 如果 m1>m2,则需要舍去序列A中较大的一半以及序列B中较小的一半,要求舍去长度相等
(注意: 如果序列中元素的个数为奇数,舍去中间点之前(或之后)的元素,保留中间点;如果是偶数,舍去中间点及中间点之前(或之后)的元素)
2、在保留的两个升序序列中,循环执行上述过程,直到两个序列中只剩下一个元素时停止循环,较小的中位数。
代码实现:
查找中位数部分
// length是序列A、序列B的长度
int MidSearch(int *A, int *B, int length){
// 设A的首数为s1, 尾数为d1, 中位数为m1;
// 设B的首数为s2, 尾数为d2, 中位数为m2;
int s1=0, d1=length-1, m1, s2=0, d2=length-1, m2;
while(s1!=d1||s2!=d2){
// 取中位数下标
m1=(s1+d1)/2; m2=(s2+d2)/2;
// 判断m1、m2的关系
if(A[m1]==B[m2]){ // 满足条件1
return A[m1];
}else if(A[m1]<B[m2]){ // 满足条件2
if((s1+d1)%2==0){
s1=m1; d2=m2;
}else{
s1=m1+1; d2=m2;
}
}else{ // 满足条件3
if((s1+d1)%2==0){
d1=m1; s2=m2;
}else{
d1=m1; s2=m2+1;
}
}
}
return A[s1]<B[s2] ? A[s1] : B[s2];
}