leetcode上面有道“Median of Two Sorted Arrays”关于求中位数的题目,时间复杂度要求为O(log (m+n)),因此正好在本文梳理下关于
求中位数的方法。
规则:
1)元素个数为奇数:中间那个
2)元素个数为偶数:中间两个数求平均数
A.首先,寻找中位数最简单直接的办法肯定是循环从头加起,而此种方法的时间复杂度显然为O(m+n)
public static double findMedianSortedArrays(int[] nums1, int[] nums2) {
int k=nums1.length+nums2.length;
if(k%2==1){
return findkth(nums1,nums2,k/2+1);
}
else{
return (findkth(nums1,nums2,k/2)+findkth(nums1,nums2,k/2+1))/2;
}
}
public static double findkth(int[] A,int[] B,int k){
int i=0,j=0,n=0;
while((i<A.length)&&(j<B.length)){
if(A[i]<B[j]){
n++;
if(n==k){
return A[i];
}
i++;
}
else{
n++;
if(n==k){
return B[j];
}
j++;
}
}
if(i<A.length){
return A[i-1+k-n];
}
else{
return B[j-1+k-n];
}
}
B.由于时间复杂度的要求,本问题很容易想到使用二分法解决,我们首先借助寻找第k大元素的方法解决。
二分法寻找第k大元素的两个核心点:
1)如果数组A的元素个数小于k/2,则第k个元素不可能出现在数组B的前k/2个元素中,也就是说我们可以将数组B的前key/2个元素
舍去
2)如果数组A的第k/2个元素小于数组B的第k/2个元素,则可以舍去数组A的前k/2个元素
即每次减少两数组任一数组前k-1个元素中的k/2个元素,直到k减少为1,或者两个数组中只有一个数组还有元素,另一个数组为空(即
B_start>=B.length)
public static double findMedianSortedArrays(int[] nums1, int[] nums2) {
int len=nums1.length+nums2.length;
if(len%2==1){
return findkth(nums1,0,nums2,0,len/2+1);
}
return (findkth(nums1,0,nums2,0,len/2)+findkth(nums1,0,nums2,0,len/2+1))/2;
}
public static double findkth(int[] A,int A_start,int[] B,int B_start,int k){
if(A_start>=A.length){//此处==也可以,因为导致越界的原因在于当某一数组去除掉其中的k/2个元素后正好为空
return B[B_start+k-1];
}
if(B_start>=B.length){
return A[A_start+k-1];
}
if(k==1){
return Math.min(A[A_start],B[B_start]);
}
int A_mid=A_start+k/2-1<A.length
?A[A_start+k/2-1]
:Integer.MAX_VALUE;
int B_mid=B_start+k/2-1<B.length
?B[B_start+k/2-1]
:Integer.MAX_VALUE;
//如果数组A的元素个数小于k/2,将数组B的前key/2个元素舍去
if(A_mid<B_mid){
return findkth(A,A_start+k/2,B,B_start,k-k/2);//注意:k-k/2不等于k/2
}
else{
return findkth(A,A_start,B,B_start+k/2,k-k/2);
}
}