[LeetCode.Trick]Median of Two Sorted Arrays

最新推荐文章于 2021-08-05 15:50:50 发布

原创最新推荐文章于 2021-08-05 15:50:50 发布 · 365 阅读

0 ·

CC 4.0 BY-SA版权

本文介绍了一种高效算法来查找两个已排序数组的中位数。该算法利用了分治策略，每次排除其中一个数组的一半数据，从而快速定位到目标元素。文章提供了完整的Java代码实现。

原题链接

一题一句：做这个题的关键是经典方法——找到俩个sorted array里面第K大的元素。我们无法通过比较A_key 和 B_key知道K在哪，但是可以确定K不在哪。每一层可以排除A的1/2或是B的1/2

Notice : k - k/2 != k/2

public class Solution {
    public static double findMedianSortedArrays(int A[], int B[]) {
        int total = A.length + B.length;
        if(total%2 == 0){
            return (findKth(A,0,B,0,total/2) + findKth(A,0,B,0,total/2+1))/2.0;
        }else{
            return findKth(A,0,B,0,total/2 + 1);
        }
    }
    public static double findKth(int[] A, int A_start, int[] B, int B_start,int k){
        if(A_start >= A.length){
            return B[B_start+k-1];
        }
        if(B_start >= B.length){
            return A[A_start+k-1];
        }
        if(k == 1){
            return Math.min(A[A_start],B[B_start]);
        }
        int AKey = A_start+k/2-1 < A.length ? A[A_start + k/2 -1] : Integer.MAX_VALUE;
        int BKey = B_start+k/2-1 < B.length ? B[B_start + k/2 -1] : Integer.MAX_VALUE;
        if(AKey > BKey){
            return findKth(A,A_start,B,B_start+k/2,k-k/2);
        }else{
            return findKth(A,A_start+k/2,B,B_start,k-k/2);
        }
    }
}