[leetcode] Median of Two Sorted Arrays

和道一文字JC

于 2013-08-16 22:32:53 发布

阅读量603

点赞数

CC 4.0 BY-SA版权

分类专栏：算法题

本文链接：https://blog.youkuaiyun.com/mungco/article/details/10012135

算法题专栏收录该内容

107 篇文章

订阅专栏

本文介绍了一种高效算法，用于找到两个已排序数组的中位数，其运行时间为O(log(m+n))。

摘要生成于 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)).

class Solution {
public:
    double findMedianSortedArrays(int A[], int m, int B[], int n) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        double midA,midB;
        midA=FindKth(A,0,m,B,0,n,(m+n)/2+1);
        if((m+n)%2==0)
            midB=FindKth(A,0,m,B,0,n,(m+n)/2);
        else
            midB=midA;
        return (midA+midB)/2;
    }
    int FindKth(int A[], int aoffset,int m , int B[], int boffset, int n, int k){
        if(m>n)
            return FindKth(B,boffset,n,A,aoffset,m,k);//巧妙的解决一个选择最长的那个数组的问题
        if(m==0) return B[k-1];
        if(k==1) return min(A[aoffset],B[boffset]);

        int pa=min(k/2,m);
        int pb=k-pa;

        if(A[pa+aoffset-1]<B[pb+boffset-1])
            return FindKth(A,aoffset+pa,m-pa,B,boffset,n,k-pa);
        return FindKth(A,aoffset,m,B,boffset+pb,n-pb,k-pb);
    }
};