这道题挺有趣的。
我百度了一个思路就是,写一个find函数,可以寻找第k大的数,复杂度是log(k).
思路就是,要寻找第k大的数,每次在两个数组的其中一个中杀掉k/2个数就行了。
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int n=nums1.length;
int m=nums2.length;
int a=(n+m+1)>>1; //左边的数
int b=(n+m+2)>>1; //右边的数
//两个数相加除以2.0
return (find(nums1,0,nums2,0,a)+find(nums1,0,nums2,0,b))/2.0;
}
//递归寻找数组的第i个元素
int find(int x[],int l,int y[],int r,int k) {
if(l>=x.length)return y[r+k-1]; //如果起始点大于等于数组长度,则对应点在另一个位置
if(r>=y.length)return x[l+k-1];
if(k==1)return Math.min(y[r], x[l]); //如果k=1,就返回最小值
int mid1=l+(k>>1)-1<x.length?l+(k>>1)-1:x.length-1; //取两个数对应点的l+k/2-1的位置,如果超过,取终点
int mid2=r+(k>>1)-1<y.length?r+(k>>1)-1:y.length-1;
if(x[mid1]<y[mid2]) { //然后那个值小,把前面的k/2个全部杀掉。从剩下的找。
return find(x,mid1+1,y,r,k-(mid1-l)-1);
}
else
return find(x,l,y,mid2+1,k-(mid2-r)-1);
}