leetcode median of two sorted arrays

找两个排好序的数组的中位数。


用快排的想法去找,发现code真不容易,写成一团了。


收藏两个别人的代码吧。。


double findMedianSortedArrays2(int A[], int m, int B[], int n) {
	if (m > n) return findMedianSortedArrays(B, n, A, m);
	int minidx = 0, maxidx = m, i, j, num1, mid = (m + n + 1) >> 1, num2;
	while (minidx <= maxidx)//实际上是找到A和B一起的前mid个数,i+j=mid,从而A[0]到[i-1]和B[0]到[j-1]一共就是mid个数
	{
		i = (minidx + maxidx) >> 1;
		j = mid - i;
		if (i<m && j>0 && B[j - 1] > A[i]) minidx = i + 1;
		else if (i>0 && j<n && B[j] < A[i - 1]) maxidx = i - 1;
		else
		{
			if (i == 0) num1 = B[j - 1];
			else if (j == 0) num1 = A[i - 1];
			else num1 = max(A[i - 1], B[j - 1]);
			break;
		}
	}
	if (((m + n) & 1)) return num1;
	if (i == m) num2 = B[j];
	else if (j == n) num2 = A[i];
	else num2 = min(A[i], B[j]);
	return (num1 + num2) / 2.;
}

int getkth(int s[], int m, int l[], int n, int k){
	// let m <= n
	if (m > n)
		return getkth(l, n, s, m, k);
	if (m == 0)
		return l[k - 1];
	if (k == 1)
		return min(s[0], l[0]);

	int i = min(m, k / 2), j = min(n, k / 2);
	if (s[i - 1] > l[j - 1])
		return getkth(s, m, l + j, n - j, k - j);
	else
		return getkth(s + i, m - i, l, n, k - i);
	return 0;
}


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值