兩個有序數組的第n大數

標籤：有序數組   中位元   

兩個有序數組，各自含有n個元素，求第n大的元素

1.順序遍曆兩個數組，計數變數k統計出現的第k小元素，時間複雜度為O(n)

代碼如下：

int getmid(int a[],int b[],int n){int k=0;int i=0,j=0;while(i<n&&j<n){if(a[i]<b[j]){i++;k++;if(k==n)return a[i-1];}else {j++;k++;if(k==n)return b[j-1];}}}

2.二分的方法

    取A數組的中間元素mid1,取B數組的中間元素mid2,先比較這兩個元素的大小，如果這兩個元素相等，則直接返回A[mid1],如果A[mid1]<B[mid2],則mid1左側的元素可以去掉，B數組右側的元素可以去掉，這裡還要區分數組元素個數為偶數奇數的情況，如果元素個數為偶數，則mid1元素也要去掉。如果A[mid1]<B[mid2]的情況與此類似。時間複雜度為O(logn)

# include <iostream># include <cstdlib>using namespace std;int mid(int a[],int b[],int n){int s1=0,e1=n-1;int s2=0,e2=n-1;int mid1=(s1+e1)/2;int mid2=(s2+e2)/2;while(s1!=e1||s2!=e2){mid1=(s1+e1)/2;mid2=(s2+e2)/2;if(a[mid1]==b[mid2]){return a[mid1];}if(a[mid1]<b[mid2]){if((s1+e1)%2==0){s1=mid1;e2=mid2;}else {s1=mid1+1;e2=mid2;}}else{if((s1+e1)%2==0){e1=mid1;s2=mid2;}else{e1=mid1;s2=mid2+1;}}}return a[s1]<b[s2]?a[s1]:b[s2];}int main(){int a[5]={2,4,5,6,9};int b[5]={1,3,7,8,10};cout<<mid(a,b,5)<<endl;system("pause");return 0;}

兩個有序數組的第n大數