求两个有序数组的中位数

最新推荐文章于 2021-04-20 23:29:17 发布

jinglinxiao

最新推荐文章于 2021-04-20 23:29:17 发布

阅读量241

点赞数

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.csdn.net/jinglinxiao/article/details/79347837

版权

leetcode 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

求两个有序数组的中位数
设用 $i$ 分割数组定义从在数组的第 $i$ 个元素的左边将数组分割成两个子数组。如 $i=2$ 时， $[1,2,3,4]→[1,2]和[3,4]$ 。
那么为了找两个有序数组的中位数，假设两个数组长度之和为偶数，我们用 $i$ 分割第一个数组， $j$ 分割第二个数组，那么 $i+j=\frac{len(arr1)+len(arr2)}{2}$ 且设 $L_i$ 为第 $i$ 个数组的割的左边的数，如果不存在则为无穷小， $R_i$ 为第 $i$ 个数组的割右边的数，如果不存在则为无穷大，那么 $L_1 \leq R_2$ ， $L_1 \leq R_1$ 满足。因为当 $i$ 确定了那么 $j$ 也确定了。因此可以用数组长度最短的那个数组进行二分，如果 $L_1 > R_2$ 说明 $i$ 还需左移，如果 $L_2>R_1$ ，说明 i <script type="math/tex" id="MathJax-Element-2143">i</script> 还需右移。
为了保证数组长度之和是偶数，可以在数组中添加 # 。

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int n = nums1.size(), m = nums2.size();
        if(n>m) return findMedianSortedArrays(nums2, nums1);// nums1最短
        int total=n+m;
        int l=0,r=2*n,L1,R1,L2,R2;
        //将原数组中加'#'，让两个数组的长度都为偶数。这样就避免了两个数组长度加起来是奇数的问题。
        //如 [1 2 3 4] 变为 [# 1 # 2 # 3 # 4]
        while(l<=r)
        {
            int c1=(l+r)/2;
            int c2=n+m-c1;
            //割到左边沿，值为INT_MIN，割到右边沿，值为INT_MAX
            L1=(c1==0)?INT_MIN:nums1[(c1-1)/2];
            R1=(c1==2*n)?INT_MAX:nums1[c1/2];
            L2=(c2==0)?INT_MIN:nums2[(c2-1)/2];
            R2=(c2==2*m)?INT_MAX:nums2[c2/2];
            if(L1>R2) r=c1-1;
            else if(L2>R1) l=c1+1;
            else break;
        }
        return (max(L1,L2)+min(R1,R2))/2.0;
    }
};