求兩個排序陣列的中位數
阿新 • • 發佈:2019-02-01
設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中找
轉載:http://www.cnblogs.com/shuaiwhu/archive/2011/04/15/2065063.html#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; }