[Leetcode] Median of Two Sorted Arrays

最新推荐文章于 2025-09-10 15:24:29 发布

转载最新推荐文章于 2025-09-10 15:24:29 发布 · 46 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/Phoebe815/p/4087312.html

文章标签：

#数据结构与算法

本文提供了一种在两个已排序数组中查找中位数的方法，整体运行时间复杂度为O(log(m+n))。通过递归查找第k个元素实现，包含详细的Java代码示例及解析。

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)).

Solution:

public class Solution {
    public double findMedianSortedArrays(int A[], int B[]) {
        int m=A.length;
        int n=B.length;
        int total=m+n;
        if(total%2!=0){
            return findKthNumber(A,m,B,n,total/2+1);
        }else{
            return (findKthNumber(A,m,B,n,total/2)+findKthNumber(A,m,B,n,total/2+1))/2;
        }
    }

    private double findKthNumber(int[] a, int m, int[] b, int n, int k) {
        // TODO Auto-generated method stub
        if(m>n)
            return findKthNumber(b, n, a, m, k);
        if(m==0)
            return b[k-1];
        if(k==1)
            return Math.min(a[0], b[0]);
        int pa=Math.min(k/2, m);
        int pb=k-pa;
        if(a[pa-1]<b[pb-1]){
            int[] c=Arrays.copyOfRange(a, pa, m);
            return findKthNumber(c, m-pa, b,n, k-pa);
        }else if(a[pa-1]>b[pb-1]){
            int[] d=Arrays.copyOfRange(b, pb, n);
            return findKthNumber(a, m, d, n-pb, k-pb);
        }else{
            return a[pa-1];
        }
    }
}