过程:
一、求中位数,最简单直接的想法就是抓一串已经完成排序的数组,通过下标访问元素,进而求出中位数。但很明确这么做,时间复杂度只能是O(nlogn),达不到题目要求的O(log(m+n))。但想了半天,实在想不出其他好办法了,就试着提交一下,结果也过了。代码(CPP)如下:
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int len=0;
nums1.insert(nums1.end(),nums2.begin(),nums2.end());
len=nums1.size();
sort(nums1.begin(), nums1.end());
if(len%2==0)
return ((double)nums1[len/2]+(double)nums1[len/2-1])/2;
else
return (double)nums1[len/2];
}
};
二、就这么过了有一种如鲠在喉的感觉,而且这种方法并没有很好地利用两个数组已经排序的条件。于是看了官方的解答。仿写了C语言版的代码如下:
double findMedianSortedArrays(int* A, int m, int* B, int n) {
int iMin=0,iMax=m,halfLen=(m +n+1)/2,i,j,maxLeft,minRight;
while (iMin <= iMax)
{
i = (iMin + iMax) / 2;
int j = halfLen - i;
if (i < iMax && B[j-1] > A[i])
iMin = i + 1;
else if (i > iMin && A[i-1] > B[j])
iMax = i - 1;
else
{
maxLeft = 0;
if (i == 0) { maxLeft = B[j-1]; }
else if (j == 0) { maxLeft = A[i-1]; }
else { maxLeft = (A[i-1]>B[j-1]?A[i-1]:B[j-1]); }
if ( (m + n) % 2 == 1 ) { return maxLeft; }
//printf("%d\n",);
minRight = 0;
if (i == m) { minRight = B[j]; }
else if (j == n) { minRight = A[i]; }
else { minRight = (B[j]>A[i]?A[i]:B[j]); }
return (maxLeft + minRight) / 2.0;
}
}
return 0.0;
}
接着悲伤的故事发生了,虽然在自己电脑上运行都正确,但每次提交,都显示:对输入为 A[]={1},B[]={2,3}的用例,输出总是0.0……(这个问题至今仍未解决,若有大佬路过,还望不吝赐教^_^)
因本人有强迫症,必须要提交成功一次C的代码,于是参考了这篇博客(感谢前辈^_^),写得通过的代码如下:
int min(int a,int b)
{
if(a>b)
return b;
return a;
}
double findKth(int a[], int m, int b[], int n, int k)
{
if (m > n)
return findKth(b, n, a, m, k);
if (m == 0)
return b[k - 1];
if (k == 1)
return min(a[0], b[0]);
int pa = min(k / 2, m), pb = k - pa;
if (a[pa - 1] < b[pb - 1])
return findKth(a + pa, m - pa, b, n, k - pa);
else if (a[pa - 1] > b[pb - 1])
return findKth(a, m, b + pb, n - pb, k - pb);
else
return a[pa - 1];
}
double findMedianSortedArrays(int A[], int m, int B[], int n)
{
int total = m + n;
if (total & 0x1)
return findKth(A, m, B, n, total / 2 + 1);
else
return (findKth(A, m, B, n, total / 2)+ findKth(A, m, B, n, total / 2 + 1)) / 2;
}