Median of Two Sorted Arrays （JAVA）

最新推荐文章于 2022-04-06 14:11:33 发布

原创最新推荐文章于 2022-04-06 14:11:33 发布 · 2.2k 阅读

CC 4.0 BY-SA版权

本文介绍了一个算法，该算法可以找到两个已排序数组的中位数。通过实现一个名为findMedianSortedArrays的方法，它能够处理不同长度的数组，并且通过递归调用findKth函数来找到特定位置的元素。此方法适用于计算机科学中的数据结构和算法领域。

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

class Solution {
public 
double findKth(int a[], int startA, int m, int b[], int startB, int n, int k)
{
	if (m > n) return findKth(b, startB, n, a, startA, m, k);
	 
    if (m == 0) 
        return b[startB + k-1];
    
    if (k == 1) 
        return min(a[startA], b[startB]);

    int pa = min(k/2, m), pb = k - pa;
    
    if (a[startA + pa-1] < b[startB + pb-1]) 
        return findKth(a, startA + pa, m-pa, b, startB, n, k - pa);
    else 
        return findKth(a, startA, m, b, startB + pb, n-pb, k-pb);
}
    public void test() {
         
     }
    private int min(int a, int b){
    	return a > b ? b : a;
    }
    public double findMedianSortedArrays(int A[], int B[]) {
        int m = A.length;
        int n = B.length;
        
        int total = m+n;
        
        if ((total&1) == 1)
            return findKth(A, 0, m, B, 0, n, total/2+1);
        else
    return (findKth(A, 0, m, B, 0, n, total/2) + findKth(A, 0, m, B, 0, n, total/2+1))/2;
    }
     
};