题目见https://blog.csdn.net/qq_35684046/article/details/104330155
法一:
我们用两个指针i,j,分别指向n1与n2
遍历(len1+len2+1)/2次后跳出循环。
pre = now;
(1)如果i<n1.lenth&&(j>=n2.lenth||n1[i]>n2[i]),令now = n1[i++]。
(2)其他情况下,now = n2[j++];
如果两数组合并后为奇数返回now,否则返回(pre+now)/2。
时间复杂度为O((m+n)/2),并没有达到题目的要求,所以我们尝试法二。
法二:
我们猜想,如果要达到题目所要求的O(log(m+n)),可能需要用到二分法。
我们始终令最短的数组为n1,这样有利于我们下面的操作。
假设我们要查找在合并后数组中第k大的元素。
(1)我们分别在两个数组内查看K/2处的元素(如果K/2大于该数组的长度,便查询该数组末尾元素)。
(2)比较这两个元素的大小,假如n1[k/2]<n2[k/2],则说明n1数组k/2处的元素一定不是总的第k大的元素。因为我们假设,n2前k/2-1个元素都小于n1第k/2个元素,那么最大也只是合并后后第k-1个元素,所以我们将n1前k/2个元素全部删除(下标变了,并没有真正的删除)。并将K减去删除掉的元素个数。
重复以上步骤,直到k=1或某个数组被删除空。
只是分享了大概的思路,请理解大概之后,配合代码使用。
java代码
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int m = nums1.length;
int n = nums2.length;
return (helper(nums1,nums2,0,0,(m+n+1)/2)+helper(nums1, nums2,0,0, (m+n+2)/2))*0.5;
}
public int helper(int[] a,int[]b,int aS,int bS, int k) {
int aE = a.length-1;
int bE = b.length-1;
int len1 = aE-aS+1;
int len2 = bE-bS+1;
System.out.println("----------------");
System.out.println(" len1:"+len1+" len2:"+len2+" k:"+k);
if(len1>len2) {
return helper(b,a,bS,aS,k);
}
if(len1==0) {
return b[bS+k-1];
}
if(k == 1) {
return Math.min(a[aS], b[bS]);
}
int i = aS+Math.min(k/2, len1)-1;
int j = bS+Math.min(k/2, len2)-1;
System.out.println("i:"+i+" j:"+j+" len1:"+len1+" len2:"+len2+" k:"+k);
System.out.println("----------------");
if(a[i]>=b[j]) {
k -= (j-bS+1);
bS = j+1;
}
else {
k -= (i-aS+1);
aS = i+1;
}
return helper(b,a,bS,aS,k);
}