首页 > 代码库 > LeetCode - Median of Two Sorted Arrays
LeetCode - Median of Two Sorted Arrays
题目描述:找出两个有序数组合并之后的中位数。复杂度控制在log(N)
如:
nums1 = [1, 2]
nums2 = [3, 4]
The median is (2 + 3)/2 = 2.5
思路:
先转成logN内求两个数组第K大,二分即可实现。
在求 第(N+2)/2 是多少,以及 (N+1)/2是多少,即可。
AC代码:
int min(int a,int b){
if(a<b)return a;
return b;
}
int fk(int* n1,int n,int* m1 ,int m,int k){
if (n < m){
return fk(m1,m,n1,n,k);
}
if(m==0){
return n1[k-1];
}
if(k==1){
return min(n1[0],m1[0]);
}
int l = min(k/2,n);
int r = min(k/2,m);
if(n1[l-1] < m1[r-1]){
return fk(n1+l,n-l,m1,m,k-l);
}else{
return fk(n1,n,m1+r,m-r,k-r);
}
}
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) {
int l = (nums1Size + nums2Size+2)/2;
int r = (nums1Size + nums2Size+1)/2;
return (fk(nums1,nums1Size,nums2,nums2Size,l)+fk(nums1,nums1Size,nums2,nums2Size,r))/2.0;
}
LeetCode - Median of Two Sorted Arrays