题目
一个长度为L(L>=1)的升序序列S,处在第(L/2)个位置的数称S的中位数。两个序列的中位数是包含他们所有元素的升序序列的中位数,试设计一个在时间和空间两方面都尽可能高效的算法,找出两个升序序列A和B的中位数。
分析
一个升序序列S1={11,13,15,17,19},S1的中位数是15,S2={2,4,6,8,20},则S1和S2的中位数是11。
处在第(L/2)位置的数,实际上是对L/2的结果向上取整,S1的长度是5,S1的中位数在第三个位置是15,S1和S2两个序列的长度为10,他们两个构成的新序列为S3={2,4,6,8,11,13,15,17,19,20},长度为10,S3的中位数在第五个位置是11。
A和B升序,如果A,B的中位数相等,那么这个数就是中位数。否则从A,B的头开始遍历,找第n个数,这个数就是A和B的中位数。
代码
int mSearch(int A[], int B[], int n) {
///两个数组, n为数组的长度
if(A[n / 2] == B[n / 2]) {///A和B的中位数相等,直接返回
return A[n / 2];
}
int k = 1;///k表示当前比较的值是第几个
int i, j;
for(i = 0, j = 0; i < n, j < n; ) {///从A、B头开始遍历,找第k个位置
if(A[i] < B[j]) {
i++;
}else {
j++;
}
k++;
if(k == n) {
return A[i] < B[j] ? A[i] : B[j];
}
}
if(i == n) {///只剩B中的数据
for(;;j++) {
k++;
if(k == n) {
return B[j];
}
}
}
if(j == n) {///只剩A中的数据
for(;;j++) {
k++;
if(k == n) {
return A[i];
}
}
}
}
源代码
http://123.206.59.223:8080/code/code/3.rar