Leetcode_4_Median of Two Sorted Arrays

过程:

一、求中位数,最简单直接的想法就是抓一串已经完成排序的数组,通过下标访问元素,进而求出中位数。但很明确这么做,时间复杂度只能是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;
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值