1. 程式人生 > >求兩個排序陣列的中位數

求兩個排序陣列的中位數

設X[1...n]和Y[1...n]為兩個陣列,每個都包含n個已排序好的數。給出一個求陣列X和Y中所有2n個元素的中位數的、O(lgn)時間的演算法。

演算法思想:

1.兩個陣列中小於median的個數為(n - 1)個,假設該median為陣列a中的第k個,k為陣列下標,那麼在陣列a中已經存在k個值小於median,那麼在陣列b中必然有(n - 1) - k = (n-k-1)個數小於median,如果b[n - k - 2] <= median <= b[n - k - 1],那麼median就找到了,如果median >= b[n - k - 1],則搜尋陣列a中[0, k-1]中的元素,如果median <= b[n - k - 2],則搜尋陣列a中[k+1, n- 1]中的元素.

2.依次類推,類似二分查詢,不斷重新設定low,high,middle

3.如果在陣列a中沒找到,則在陣列b中找

#include<iostream>
using namespace std;

int Find(int *a,int *b,int n)
{
	if(a==NULL||b==NULL||n<=0)
	{
		cout<<"error";
		return 0;
	}
	int low=0,high=n-1;
	while(low<=high)
	{
		int mid=(low+high)/2;
		if(mid==n-1)                     //到達邊界
		{
			if(a[mid]<b[0])
				return a[mid];
			else                    //說明中位數一定位於陣列b中。
				return -1;
		}
		else
		{
			if(a[mid]>=b[n-mid-2]&&a[mid]<=b[n-mid-1])
				return a[mid];
			else if(a[mid]>b[n-mid-1])
				high=mid-1;
			else
				low=mid+1;
		}
	}
	return -1;
}

int main()
{
	int a[]={1,3,4,6,7};
	int b[]={2,5,8,9,10};
	if(Find(a,b,5)==-1)
		cout<<Find(b,a,5)<<endl;
	else
		cout<<Find(a,b,5)<<endl;
	return 0;
}
轉載:http://www.cnblogs.com/shuaiwhu/archive/2011/04/15/2065063.html