[leetcode 4] Median of Two Sorted Arrays

最新推荐文章于 2025-09-13 14:08:34 发布

「已注销」

最新推荐文章于 2025-09-13 14:08:34 发布

阅读量227

点赞数

CC 4.0 BY-SA版权

分类专栏： [LeetCode 刷题] 文章标签： leetcode Binary Search

本文链接：https://blog.youkuaiyun.com/xinxinxv/article/details/42327033

[LeetCode 刷题] 专栏收录该内容

158 篇文章

订阅专栏

本文介绍了一个算法问题，即找到两个已排序数组的中位数，并提供了一个运行复杂度为O(log(m+n))的解决方案。通过递归地查找第k小元素的方法实现了这一目标。

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) {
        int total = m + n;
        if (total%2) {
            return findKthValue(A,m,B,n,total/2+1);
        } else {
            return (findKthValue(A,m,B,n,total/2) + findKthValue(A,m,B,n,total/2+1))/2.0;
        }
    }
    double findKthValue(int A[], int m, int B[], int n, int k) {
        if (m > n) {
            return findKthValue(B, n, A, m, k);
        }
        if (m == 0) {
            return B[k-1];
        }
        if (k == 1) {
            return min(A[0],B[0]);
        }
        int a = min(m, k/2);
        int b = k - a;
        if (A[a-1] == B[b-1]) {
            return A[a-1];
        } else if (A[a-1] < B[b-1]) {
            return findKthValue(A+a, m-a, B, n, k-a);
        } else {
            return findKthValue(A, m, B+b, n-b, k-b);
        }
    }
};