There are two sorted arrays A and B of size m and n respectively. Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).
方法1:将两个有序数组合并成一个有序数组,若合并后的有序数组的长度为奇数则取中间数;否则取中间两个的平均值。但这样做效率较低,达到O((m+n)log(m+n))。
double findMedianSortedArrays(int A[],int m,int B[],int n)
{
int i;
vector<double> v;
for(i=0;i<m;i++)
v.push_back(A[i]);
for(i=0;i<n;i++)
v.push_back(B[i]);
sort(v.begin(),v.end());
return double(((m+n)%2)?v[(m+n)/2]:(v[(m+n)/2]+v[(m+n+1)/2])/2.0);
}
double findMedianSortedArrays(int A[],int m,int B[],int n)
{
bool flag=((m+n)%2)?true:false;//true is odd
int i,j,step;
i=0,j=0;
if(m&&n)
step=2;
else if(m||n)
step=1;
else
return 0;
while(step<=(m+n)/2)
{
if(m==0)
{
j++;
}else if(n==0)
{
i++;
}else
{
if(A[i]<B[j])
i++;
else
j++;
}
step++;
}
if(flag)
return double(max(A[i],B[j]));
else
{
if(A[i]>B[j])
{
if(i>0)
return double((A[i]+max(A[i-1],B[j]))/2.0);
else
return double((A[i]+B[j])/2.0);
}else if(B[j]>A[i])
{
if(j>0)
return double((B[j]+max(B[j-1],A[i]))/2.0);
else
return double((A[i]+B[j])/2.0);
}
}
}
Submission Result: Wrong Answer
Input: [], [2,3]
Output: 69328905.50000
Expected: 2.50000
有哪位大神能指出错在哪里吗????
方法3:Kth element in 2 sorted array的拓展
O(n)的解法比较直观,直接merge两个数组,然后求中间值。而对于O(log(m+n))显然是用二分搜索了, 相当于“Kth element in 2 sorted array”的变形。如果(m+n)为奇数,那么找到“(m+n)/2+1 th elementin 2 sorted array”即可。如果(m+n)为偶数,需要找到(m+n)/2 th 及(m+n)/2+1 th,然后求平均。
而对于“Kth element in 2 sorted array”, 如下图,两个中位数 A[m/2] 和 B[n/2], 可以将数组划分为四个部分。而丢弃哪一个部分取决于两个条件:1, (m/2 + n/2)?k;2,A[m/2] ? B[n/2];
简单的说,就是或者丢弃最大中位数的右区间,或者丢弃最小中位数的左区间。
int GetMedian(int a[],int n,int b[],int m,int k)//在有序数组a,b中查找第k个数,这个函数可以作为一个模板
{
//assert(a&&b);
if(n<=0)
return b[k-1];
if(m<=0)
return a[k-1];
if(k<=1)
return min(a[0],b[0]);
if(b[m/2]>=a[n/2])
{
if((n/2+m/2+1)>=k)//只需在a和b的前半部找
return GetMedian(a,n,b,m/2,k);
else//只需在a的后半部和b中找
return GetMedian(a+n/2+1,n-(n/2+1),b,m,k-(n/2+1));
}else
{
if((m/2+n/2+1)>=k)//只需在b和a的前半部找
return GetMedian(a,n/2,b,m,k);
else//只需在b的后半部和a中找
return GetMedian(a,n,b+m/2+1,m-(m/2+1),k-(m/2+1));
}
}
double findMedianSortedArrays(int A[],int m,int B[],int n)
{
if((m+n)%2==0)//为偶数
return (GetMedian(A,m,B,n,(m+n)/2)+GetMedian(A,m,B,n,(m+n)/2+1))/2.0;
else
return GetMedian(A,m,B,n,(m+n)/2+1);
}