原题地址
题目内容
题目分析
看到题目第一眼想到的就是采用merge的做法,将两个数列从第一个开始比较排序,不过这样的做法复杂度就变为O(n),不符合题目的要求。后面参考了leetcode上面的discuss以及别人的博客,就采用了分治的做法,主要思想就是求数列中第k大的数。
举个例子
代码实现
double getkth(int* nums1, int size1, int* nums2, int size2,int k){
if(size1 == 0){
return nums2[k-1];
}
if(size2 == 0){
return nums1[k-1];
}
if(k == 1){
return nums1[0]<nums2[0]?nums1[0]:nums2[0];
}
int i = size1<(k/2)?size1:(k/2);
int j = size2<(k/2)?size2:(k/2);
if(nums1[i-1] > nums2[j-1]){
return getkth(nums1,size1,nums2+j,size2-j,k-j);
}
else return getkth(nums1+i,size1-i,nums2,size2,k-i);
}
double findMedianSortedArrays(int* nums1, int nums1Size, int* nums2, int nums2Size) {
int tot = nums1Size + nums2Size;
if(tot % 2 == 0){
return (getkth(nums1, nums1Size, nums2, nums2Size,tot/2)+getkth(nums1,nums1Size,nums2,nums2Size,tot/2+1))/2.0;
}else{
return getkth(nums1, nums1Size, nums2, nums2Size,tot/2+1);
}
}