Leetcode日记（4）---Median of Two Sorted Arrays

最新推荐文章于 2020-12-28 12:16:47 发布

原创最新推荐文章于 2020-12-28 12:16:47 发布 · 378 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

数据结构/算法专栏收录该内容

17 篇文章

订阅专栏

本文介绍了一种在O(log(m+n))的时间复杂度内找到两个已排序数组中位数的方法。通过递归地将问题规模减半，实现了高效的解决方案。文中提供了一个C++实现的例子。

There are two sorted arrays nums1 and nums2 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(vector<int>& nums1, vector<int>& nums2) {

    }
};

class Solution {
public:
    int kth(int a[], int m, int b[], int n, int k) {
        if (m < n) return kth(b,n,a,m,k);
        if (n==0) return a[k-1];
        if (k==1) return min(a[0],b[0]);

        int j = min(n,k/2);
        int i = k-j;
        if (a[i-1] > b[j-1]) return kth(a,i,b+j,n-j,k-j);
        return kth(a+i,m-i,b,j,k-i);
    }

    double findMedianSortedArrays(int a[], int m, int b[], int n) {
        int k = (m+n)/2;
        int m1 = kth(a,m,b,n,k+1);
        if ((m+n)%2==0) {
            int m2 = kth(a,m,b,n,k);
            return ((double)m1+m2)/2.0;
        }
        return m1;
    }
};