给定两个大小为 m 和 n 的正序(从小到大)数组 nums1 和 nums2。
请你找出这两个正序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。
你可以假设 nums1 和 nums2 不会同时为空。
示例 1:
nums1 = [1, 3]
nums2 = [2]
则中位数是 2.0
示例 2:
nums1 = [1, 2]
nums2 = [3, 4]
则中位数是 (2 + 3)/2 = 2.5
借鉴大佬得思想,感觉不是很懂:
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int len1 = nums1.length;
int len2 = nums2.length;
int len = len1 + len2;
int start1 = 0,start2 = 0;
int left = -1,right = -1;
for(int i=0;i<=len/2;i++){
left = right;//left是在循环之前赋值,那么就是得到了最后一次之前的值,然后又可以得到最后一次的值,然后进行left+right /2.0
if(start1<len1 && (start2>=len2 || nums1[start1]<nums2[start2])){
right = nums1[start1++];
}else{
right = nums2[start2++];
}
}
if((len&1)==0){
return (left+right)/2.0;
}else{
return right;//如果只有1个数的话,也就是其中一个为null,另一个为1个数。。总长度为1
}
}
}
class Solution {
public static double findMedianSortedArrays(int[] nums1, int[] nums2){
int len1 = nums1.length;
int len2 = nums2.length;
int size = len1 + len2;
if(size % 2 == 1)//奇数
return findKth(nums1, 0, len1, nums2, 0, len2, size/2 + 1); //奇数找一个值
else//偶数找2个值
return (findKth(nums1, 0, len1, nums2, 0, len2, size/2)+
findKth(nums1, 0, len1, nums2, 0, len2, size/2 + 1))/2;
}
public static double findKth(int[] nums1, int start1, int len1,
int[] nums2, int start2, int len2, int k){
//如果nums1大于nums2得长度,交换一下顺序
if(len1-start1 > len2-start2)
return findKth(nums2, start2, len2, nums1, start1, len1, k);
//如果nums1已经找完了,就找nums2得第k小得数
if(len1 - start1 == 0)
return nums2[k-1];
//第1小得数,一定是nums1和nums2得start得数
if(k == 1)
return Math.min(nums1[start1], nums2[start2]);
//p1,p2记录当前需要比较的那个位
int p1 = start1 + Math.min(len1 - start1, k/2);
int p2 = start2 + k - p1 + start1;
if(nums1[p1-1] < nums2[p2-1])
return findKth(nums1, p1, len1, nums2, start2, len2, k - p1 + start1);
else if(nums1[p1-1] > nums2[p2-1])
return findKth(nums1, start1, len1, nums2, p2, len2, k - p2 + start2);
else
return nums1[p1 -1];
}
}