Median of Two Sorted Arrays

最新推荐文章于 2023-10-07 04:23:19 发布

原创最新推荐文章于 2023-10-07 04:23:19 发布 · 615 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#Java #LeetCode #数组

LeetCode 专栏收录该内容

150 篇文章

订阅专栏

本文介绍了一种在O(log(m+n))的时间复杂度内找到两个有序数组A和B中位数的方法。通过将问题转化为寻找第k大的元素，并提供了一个具体的实现示例。

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

题目

There are two sorted arrays A and B 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)).

方法

转换为寻找第k大的数。

参考：http://blog.youkuaiyun.com/zxzxy1988/article/details/8587244

	private int findValue(int A[], int leftA, int[] B, int leftB, int k) {
		if (A.length - leftA > B.length - leftB) {
			return findValue(B, leftB, A, leftA, k);
		} 
		if (A.length - leftA == 0) {
			return B[k - 1];
		}
		if (k == 1) {
			return Math.min(A[leftA], B[leftB]);
		}
		int a = Math.min(k / 2, A.length - leftA);
		int b = k - a;
		
		if (A[a + leftA - 1] == B[b + leftB - 1]) {
			return A[a + leftA - 1];
		} else if (A[a + leftA - 1] > B[b + leftB - 1]){
			return findValue(A, leftA, B, leftB + b, k - b);
		} else {
			return findValue(A, leftA + a, B, leftB, k - a);
		}
	}
    public double findMedianSortedArrays(int A[], int B[]) {
    	int lenA = A.length;
    	int lenB = B.length;
    	int sum = lenA + lenB;
    	if ((sum & 1) == 1) {
    		return findValue(A, 0, B, 0, sum / 2 + 1);
    	} else {
    		return (findValue(A, 0, B, 0, sum / 2) + findValue(A, 0, B, 0, (sum/2 + 1))) / 2.0;
    	}
    }