[LeetCode]Median of Two Sorted Arrays 二分查找兩個有序數組的第k數(中位數)
阿新 • • 發佈:2017-06-21
大於 data div ble 關系 操作 spa 兩個 -1
二分。情況討論
因為數組有序,所以能夠考慮用二分。通過二分剔除掉肯定不是第k位數的區間。如果數組A和B當前處理的下標各自是mid1和mid2。則
1、假設A[mid1]<B[mid2],
①、若mid1+mid2+2==k(+2是由於下標是從0開始的),則
mid1在大有序數組中下標肯定小於k,所以能夠排除[0,mid1]。此外。B[mid2]下標大於或等於k。能夠排除[mid2+1,n];
②、若mid1+mid2+2<k,則
mid1在大有序數組中下標肯定小於k,所以能夠排除[0,mid1]
③、若mid1+mid2+2>k,則
B[mid2]下標大於k,能夠排除[mid2,n];
2、假設A[mid1]<B[mid2]情況相符,僅僅是下標改變。
這些操作處理完後。可能一個數組被排除了,即滿足lowX>highX。此時僅僅需對還有一個數組進行二分,同一時候二分其元素在還有一個數組中的下標,確定全局下標,終於通過推斷全局下標與k的關系。確定是否為第k數
class Solution { public: int findPos(int* p,int n,int x){ int low=0,high=n-1,mid; while(low<=high){ mid=(low+high)>>1; if(p[mid]<=x)low=mid+1; else high=mid-1; } return low; } double findK(int a[], int m, int b[], int n,int k){ int mid1,mid2,low1=0,low2=0,high1=m-1,high2=n-1,x; while(low1<=high1&&low2<=high2){ mid1=(high1+low1)>>1; mid2=(high2+low2)>>1; if(a[mid1]<b[mid2]){ if(mid1+mid2+2==k){ low1=mid1+1; high2=mid2; } else if(mid1+mid2+2<k){ low1=mid1+1; } else high2=mid2-1; } else{ if(mid1+mid2+2==k){ low2=mid2+1; high1=mid1; } else if(mid1+mid2+2<k){ low2=mid2+1; } else high1=mid1-1; } } if(low1<=high1){ // if(low1==high1)return a[low1]; while(low1<=high1){ mid1=(low1+high1)>>1; x=findPos(b,n,a[mid1]); if(x+mid1+1==k)return a[mid1]; else if(x+mid1<k)low1=mid1+1; else high1=mid1-1; } return low1>=m?a[m-1]:a[low1]; } else { // if(low2==high2)return b[low2]; while(low2<=high2){ mid2=(low2+high2)>>1; x=findPos(a,m,b[mid2]); if(x+mid2+1==k)return b[mid2]; else if(x+mid2<k)low2=mid2+1; else high2=mid2-1; } return low2>=n?a[n-1]:b[low2]; } } double findMedianSortedArrays(int a[], int m, int b[], int n) { int k=m+n; if(k&1){ return findK(a,m,b,n,k/2+1); } else{ return (findK(a,m,b,n,k/2)+findK(a,m,b,n,k/2+1))/2.0; } } };
[LeetCode]Median of Two Sorted Arrays 二分查找兩個有序數組的第k數(中位數)