leetcode:4. Median of Two Sorted Arrays(Java实现)

本文介绍了一个LeetCode上的经典算法问题——如何找到两个已排序数组的中位数,且要求整体运行时间复杂度为O(log(m+n))。通过提供的Java代码示例,详细解释了使用分治策略来解决这个问题的方法。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

leetcode测试地址:https://leetcode.com/problems/median-of-two-sorted-arrays/#/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)).

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

Subscribe to see which companies asked this question.

Java code:

package go.jacob.day617;

public class Demo2 {
	public double findMedianSortedArrays(int[] A, int[] B) {
		int m = A.length, n = B.length;
		// 不论总数是奇数还是偶数,以l和r为下标的两数的均值都是medium
		int l = (m + n + 1) / 2;
		int r = (m + n + 2) / 2;

		return (getkth(A, 0, B, 0, l) + getkth(A, 0, B, 0, r)) / 2.0;
	}

	private int getkth(int[] A, int aStart, int[] B, int bStart, int k) {
		if (aStart >= A.length)
			return B[bStart + k - 1];
		if (bStart >= B.length)
			return A[aStart + k - 1];
		if(k==1)
			return Math.min(A[aStart], B[bStart]);
		int aMin=Integer.MAX_VALUE,bMin=Integer.MAX_VALUE;
		if(aStart+k/2-1<A.length)
			aMin=A[aStart+k/2-1];
		if(bStart+k/2-1<B.length)
			bMin=B[bStart+k/2-1];
		
		if(aMin<bMin)
			return getkth(A,aStart+k/2,B,bStart,k-k/2);
		else
			return getkth(A,aStart,B,bStart+k/2,k-k/2);
	}
}




评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值