中兴笔试题:两有序数组的中位数求解

题目:设X[1..n]和Y[1..n]为两个数组,每个都包含n个已排好序的数。给出一个求数组X和Y中所有2n个元素的中位数的、O(lgn)时间的算法。
        首先假设中位数等于m,在数组X中。假设X[k]=m,于是X中就有k个元素小于等于m,n-k个元素大于等于m。我们知道如果两个数组合并起来,必然有n个元素小于等于m,n个元素大于等于m。所以数组Y中一定有n-k个元素小于等于m,n-(n-k)=k个元素大于等于m。

       所以,我们可以通过检测Y[n-k]<=X[k]<=Y[n-k+1]是否成立来确定X[k]是否是中位数。k=n是一种边界情况,这时n-k=0,不存在Y[0],于是我们只需要检测X[n]<=Y[1]是否成立。

        如果中位数在数组X中但不是X[k],上述条件将不再成立。如果中位数是X[k'] (k'<k),则X[k]大于中位数,X[k]>Y[n-k+1]。相反,如果中位数是X[k''] (k''>k),则X[k]小于中位数,X[k]<Y[n-k]。
        于是,我们可以通过使用二分查找确定是否存在X[k]使得(k<n and Y[n-k]<=X[k]<=Y[n-k+1] or k=n and X[k]<=Y[1])成立。如果找到了这样的X[k],则它就是中位数;如果没找到,则中位数在数组Y中,我们则需要用二分查找寻找Y[k]使得(k<n and X[n-k]<=Y[k]<=X[n-k+1] or k=n and Y[k]<=X[1]),而这个Y[k]就是中位数。二分查找运行时间为O(lgn),所以总共花费的时间也为O(lgn)。

FIND-MEDIAN(A, B, n, low, high)
if low>high
    then return NOT-FOUND
    else k<-(low+high)/2
         if k=n and A[n]<=B[1]
             then return A[n]
         elseif k<n and B[n-k]<=A[k]<=B[n-k+1]
             then return A[k]
         elseif A[k]>B[n-k+1]
             then return FIND-MEDIAN(A, B, n, low, k-1)
             else return FIND-MEDIAN(A, B, n, k+1, high)

TWO-ARRAY-MEDIAN(X, Y)
n<-length[X]
median<-FIND-MEDIAN(X, Y, n, 1, n)
if median=NOT-FOUND
    then median<-FIND-MEDIAN(Y, X, n, 1, n)
return median

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值