(leetcode)Median of Two Sorted Arrays

最新推荐文章于 2023-06-12 16:30:30 发布

HIT-Lori

最新推荐文章于 2023-06-12 16:30:30 发布

阅读量805

点赞数

CC 4.0 BY-SA版权

分类专栏： C/C++/C#/python ACM leetcode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/zhangyuehuan/article/details/24378137

C/C++/C#/python 同时被 3 个专栏收录

27 篇文章

订阅专栏

ACM

18 篇文章

订阅专栏

leetcode

7 篇文章

订阅专栏

本文介绍了一种寻找两个有序数组中位数的方法，通过合并两个数组并确保算法的时间复杂度为O(log(m+n))。代码示例使用C++实现。

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

Question:

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

code :

class Solution {
public:
    double findMedianSortedArrays(int A[], int m, int B[], int n) {

    vector<int> C ;
    int i=0,j=0;
    for(;i<m&&j<n;)
    {
        if(A[i]<B[j])
        {
            C.push_back(A[i]);
            i++;
        }
        else if(A[i]>B[j])
        {
            C.push_back(B[j]);
            j++;
        }
        else
        {
            C.push_back(A[i]);
            i++;
            C.push_back(B[j]);
            j++;
        }

    }
    if(i==m)
    {
        for(;j<n;j++)
            C.push_back(B[j]);
    }
    if(j==n)
    {
         for(;i<m;i++)
            C.push_back(A[i]);
    }
    double med; 
    int mid=(m+n)/2;
    if((m+n)%2==0)
    med = (double)(C[mid]+C[mid-1])/2;
    else
    med = C[mid];
    return med;
}
};