题目:
题目是给我们两个有序数组,然后合并后求它的中位数。我们可以创建一个新的数组,来储存他们合并后的有序数组,再来求得新数组的中位数就可以了。如果长度是奇数,那中位数就是最中间那个数,如果是偶数,则是中间两个数和的一半。
代码:
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
double[] nums3=new double[nums1.length+nums2.length];
int len3=nums3.length;
int i=0,j=0,z=0;
while(i<nums1.length&&j<nums2.length){
if(nums1[i]<nums2[j]){
nums3[z++]=nums1[i++];
}else{
nums3[z++]=nums2[j++];
}
}
while(i<nums1.length){
nums3[z++]=nums1[i++];
}
while(j<nums2.length){
nums3[z++]=nums2[j++];
}
if(len3%2==1){
return nums3[len3/2];
}else{
return (nums3[len3/2-1]+nums3[len3/2])/2;
}
}
}
但是,这个方法时间复杂度为O(m+n),但题目要求的复杂度为O(log (m+n)),所以我们要找其他方法。求中位数,相当于就是求第k小的数,我们就可以使用二分法来解决此题,每次排除比第k小的数小的数字,最后求得第k小的数字,也就是中位数。整体用的是递归的方法。
代码:
class Solution {
public double findMedianSortedArrays(int[] nums1, int[] nums2) {
int n=nums1.length;
int m=nums2.length;
// 找出我们要找的中位数的坐标,如果是偶数个数,则就是求他们和的一半,如果是奇数个那就是同一个坐标,即left=right
int left=(n+m+1)/2;
int right=(n+m+2)/2;
return (getK(nums1,0,n-1,nums2,0,m-1,left)+getK(nums1,0,n-1,nums2,0,m-1,right))*0.5;
}
public int getK(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 getK(nums2,start2,end2,nums1,start1,end1,k);
}
// 如果较短数组中已经没有了元素,则直接第二个数组中直接按顺序获取第k个数
if(len1==0){
return nums2[start2+k-1];
}
// 如果k=1,就找出两个数组第一个数中的最小数
if(k==1){
return Math.min(nums1[start1],nums2[start2]);
}
// 如果以上情况都不是,那我们就要开始考虑截取长度(k/2)和当前数组所剩长度的问题,应该取最小值(如果取大的数组会越界)
int i=start1+Math.min(len1,k/2)-1;
int j=start2+Math.min(len2,k/2)-1;
// 如果nums1[i]>nums2[j],则证明nums1[j]包括它之前的数都比nums1[i]小,肯定不是第k小数,则可以直接排除,则nums2下次就从j+1开始,而因为已经排除了j-start2+1个数了,则下一次递归,只需要找出第k-(j-start2+1)小的数了。nums2[j]>nums1[i]同理
if(nums1[i]>nums2[j]){
return getK(nums1,start1,end1,nums2,j+1,end2,k-(j-start2+1));
}else{
return getK(nums1,i+1,end1,nums2,start2,end2,k-(i-start1+1));
}
}
}