时间和空间复杂度最少的求两个升序数组的中位数

2019独角兽企业重金招聘Python工程师标准>>> hot3.png

    题目描述:
             一个长度为L的升序S,处在L/2向下取整处的位置的数称为S的中位数。
             现求两个等长的两个升序序列A和B的中位数。 

    分析:当然这题如果不要求最高效的话,可以完全重新开辟一个2倍长度的数组C,然后将A,B数组合并到C中,然后得其(L/2 - 1)处的值,即可。但空间复杂度较高为O(n),并且合并的时间复杂也为O(n),所以自己又进一步改进了一下,将空间复杂度降到了O(1),但时间复杂度依旧是O(n).下面给出代码:

#include
int M_serach(int s1[],int s2[]){
	int mid=0;
	int len = 5;
	int i=0,j=0;
	while(is2[j]) {
			j++;
			mid++;
			if(mid == len-1) printf("%d\n",s1[i]);
		}else{
			i++;
			mid++;
			if(mid == len-1) {
				printf("%d\n",s2[j]);
			}
		}
	}
}
//为了方便,我就任意取了几组测试数据
int s1[] = {1,3,5,7,9};
int s2[] = {2,4,6,8,20};
int main(){
	M_serach(s1,s2);
	return 0; 
} 

可以进一步优化时间复杂度,主要是借助二分的思想,算法的基本设计思想如下:分别求两个升序的序列A,B的中位数,设为a和b,求序列A,B的中位数过程如下:

  1. 若a=b,则a或b即为所求中位数,算法结束。
  2. 若a
  3. 若b>a,则舍弃序列B中较小的一半,同时舍弃序列A中较大的一半,要求两次舍弃的长度相等。

依次再保留的序列中重复上述过程,直到两个序列中均只含一个元素时为止,较小者即为所求的中位数。

int M_Search(int A[],int B[],int n){
	int s1=0;d1=n-1,m1,s2=0,d2=n-1,m2;
	//分别表示序列A和B的首位数,末尾数和中位数。
	while(s1!=d1||s2!=d2){//s1==d1&&s2==d2
		m1 = (s1+d1)/2;
		m2 = (s2+d2)/2;
		if(A[m1]==B[m2]) return A[m1];
		if(A[m1](d1-s1)%2==0奇数又因
                               // 为 (s1+s1+d1-s1+1)%2等价即(s1+d1+1)%2 ==>(s1+d1)%2==0为奇数) 
				s1 = m1;      //舍弃A中间点以前的部分且保留中间点 
				d2 = m2;	 //舍弃B中间点后面的部分且保留中间点 
			}else{			//元素个数为偶数个
				s1 = m1+1; //舍弃A中间点及中间点以前的部分 
				d2 = m2;  //舍弃B中间点以后部分且保留中间点 
			}
		}else{
			if((s2+d2)%2==0){  //元素个数为奇数个 
				s2 = m2;      //舍弃B中间点以前的部分且保留中间点 
				d1 = m1;	 //舍弃A中间点后面的部分且保留中间点 
			}else{			//元素个数为偶数个
				s2 = m2+1; //舍弃B中间点及中间点以前的部分 
				d1 = m1;  //舍弃A中间点以后部分且保留中间点 
			}
		} 
	} 
	return A[s1]

可以手动模拟一下上述算法,为什么最后去较小者,是因为我们的中位数取得是L/2向下取整的位置,最后的时间复杂为O(logn),空间复杂度为O(1).

参考资料:王道数据结构。

转载于:https://my.oschina.net/u/2484601/blog/808302

你可能感兴趣的:(时间和空间复杂度最少的求两个升序数组的中位数)