题目
代码分析
要达到有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次