description:
There are two sorted arrays nums1 and nums2 of size m and n respectively.
Find the median of the two sorted arrays. The overall run time complexity should be O(log (m+n)).
You may assume nums1 and nums2 cannot be both empty.
Example 1:
nums1 = [1, 3]
nums2 = [2]
The median is 2.0
Example 2:
nums1 = [1, 2]
nums2 = [3, 4]
The median is (2 + 3)/2 = 2.5
题解:
①看到这个题的第一想法是采用归并算法解决,将两个有序数组归并成一个,奇数个元素时取第(m+n)/2个元素的值即为所求值,偶数个则去(m+n)/2-1和(m+n)/2个平均值.速度是m+n
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) {
int nums[10000],numsSize=nums1Size+nums2Size;
int i=0,j=0,k=0;
//进行数组的有序合并
for(i=0;j<nums1Size&&k<nums2Size;i++){
if(nums1[j]>nums2[k]) nums[i]=nums2[k++];
else nums[i]=nums1[j++];
}
//合并剩余的部分
while(j<nums1Size) nums[i++]=nums1[j++];
while(k<nums2Size) nums[i++]=nums2[k++];
//合并完成
/*注意这个地方必须除以2.0,因为定义的是整形,只有这样才可以保证转化成double*/
if(numsSize%2==0) return (nums[numsSize/2]+nums[numsSize/2-1])/2.0;
else return nums[numsSize/2];
}
②第二中方法是采用分治法解决,可以想到求两个有序数组合并后的中位数,设这个位置是K,则K=(m+n)/2;取A,B数组的K/2位置的元素进行比较。有三种情况,a:A[K/2]>B[K/2],相当于合并后,B[0...K/2]个元素内不包含合并后的K位置的元素,可以舍去,从而加快效率。b:A[K/2]<B[K/2],相当于合并后,A[0...K/2]个元素内不包含合并后的K位置的元素.c:两者相等,返回其中之一即可。时间复杂度为O(m+n),因为用到了递归,所有会用到O(m+n)的递归栈
int findK(int *nums1,int size1,int *nums2,int size2,int k){
int parta=(k/2<size1?k/2:size1);
int partb=k-parta;
if(size2>size1)
return findK(nums1,size1,nums2,size2,k);
if(size1==0) return nums2[k-1];//nums1是一个空数组
if(k==1) return (nums1[0]>nums2[0]?nums2[0]:nums1[0]);//两个数组长度均为1,返回较小的一个
if(nums1[parta-1]>nums2[partb-1])
return findK(nums1,size1,nums2+partb,size2-partb,k-partb);
else if(nums1[parta-1]<nums2[partb-1])
return findK(nums1+parta,size1-parta,nums2,size2,k-parta);
else return nums1[parta-1];
}
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) {
int total = nums1Size + nums2Size;
if(total & 0x1) //奇数
return findK(nums1, nums1Size, nums2, nums2Size, total/2 + 1);
else //偶数
return 0.5 * (findK(nums1, nums1Size, nums2, nums2Size, total/2) +
findK(nums1, nums1Size, nums2, nums2Size, total/2 + 1));
}
第二种解法是看到了是看到了某个博友的贴学习而来,附上他的原文地址: https://blog.csdn.net/banana1006034246/article/details/75305634