leetcode 150 Median of Two Sorted Arrays

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);
}

方法2：采用两个指针分别指向数组A，B，依次比较当前A[i]，B[j]的大小，当走完(m+n)/2步时再讨论奇偶。

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];

如果 (m/2 + n/2) > k，那么意味着，当前中位数取高了，正确的中位数要么在 Section 1或者Section3中。如果A[m/2] > B[n/2], 意味着中位数肯定不可能在Section 2里面，那么新的搜索可以丢弃这个区间段了。同理可以推断出余下三种情况，如下所示：

If (m/2+n/2+1) > k &&  a m/2   >  b n /2  , drop  Section  2

If (m/2+n/2+1) > k && a m/2   <  b n /2  , drop   Section  4

If  (m/2+n/2+1)  <  k && a m/2  >  b n /2 ,    drop  Section 3

If (m/2+n/2+1)  <  k && a m/2   <  b n /2  ,   drop  Section  1

简单的说，就是或者丢弃最大中位数的右区间，或者丢弃最小中位数的左区间。

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);
}