Leetcode 4 寻找两个有序数组的中位数(还有一种解法没有学习)
题目
代码分析
要达到有log的时间复杂度,可以想到一定和二分有关系,我们的普通做法就是每次都从两个数组的头出去一个,当出去总量的一半之后,就完成了。
这里我们每次都出一半,具体的,如果两个数组的总量是14,我们要找了就是第7小和第8小的数字,对于第7小的,我们首先要找第7/2=3小的数字,这就是这道题的思路。例子:
https://leetcode-cn.com/problems/median-of-two-sorted-arrays/solution/xiang-xi-tong-su-de-si-lu-fen-xi-duo-jie-fa-by-w-2/
中的解法3
代码实现
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int m=nums1.length;
int n=nums2.length;
int index1=(m+n+1)/2;
int index2=(m+n+2)/2;
return (f(nums1,0,m-1,nums2,0,n-1,index1)+f(nums1,0,m-1,nums2,0,n-1,index2))*0.5;
}
public int f(int[] nums1,int start1,int end1,int[] nums2,int start2,int end2,int k)
{
int len1=end1-start1+1;
int len2=end2-start2+1;
if(len1>len2)
{
return f(nums2,start2,end2,nums1,start1,end1,k);
}
if(len1==0)
{
return nums2[start2+k-1];
}
if(k==1)
{
return Math.min(nums1[start1],nums2[start2]);
}
int index1=start1+Math.min(k/2,len1)-1;
int index2=start2+Math.min(k/2,len2)-1;
if (nums1[index1] > nums2[index2]) {
return f(nums1, start1, end1, nums2, index2 + 1, end2, k - index2 + start2 - 1);
} else {
return f(nums1, index1 + 1, end1, nums2, start2, end2, k - index1 + start1 - 1);
}
}
}学习情况
1次


查看14道真题和解析