LeetCode(C++):寻找两个有序数组的中位数

题目描述:

给定两个大小为 m 和 n 的有序数组 nums1 和 nums2

请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。

你可以假设 nums1 和 nums2 不会同时为空。

示例 1:

nums1 = [1, 3]
nums2 = [2]

则中位数是 2.0

示例 2:

nums1 = [1, 2]
nums2 = [3, 4]

则中位数是 (2 + 3)/2 = 2.5

详细思路以及举例:(转载)

1、因为是两个有序数组,所以需要两个割点C1,C2。若将两个数组(都是升序排序的数组)分别在一个割点处分为两半,那么L1(数组前半部分的最右边的元素)一定是小于R1(数组后半部分最左边的元素),同理L2

因为要找两个有序数组的中位数(为了方便奇偶性的区分,我们将两个数组分别虚拟设置为奇数,所以两个数组看为一个数组的时候,元素个数为偶数,中位数=(m1+m2)/2,此时只要找到m1,m2中位数便可以求出),若割点刚好处于中间,此时只需要找到当L1

同理当R1>L2,R2>L1时候,比较R1和R2的数值,找出最小的(因为右半部分从左到右依次增大,那么找出最靠近左边的那个数m2)。此时(m1+m2)/2便是要求的中位数。

2、首先一个数组要求中位数,必须是排序好,而且要分奇偶,才能求出中位数。现在是两个数组,奇偶性都在不确定,所以原博主很明智地将数组进行了处理,通过虚拟地加入#符号,使得数组恒为奇。(例如:1,2,48,90  虚拟加#符号后变为:#1#2#48#90#),此处的处理是通过公示:2n+1始终是奇数而来的。代码中并没有实际加入符号#,只是将数组的长度进行了2n+1的翻倍。

3、所以只要找到处于中间的那个割点K,并找到K左右的两个数值,就可以求出中位数。对于单个有序数组,K的位置就是在数组长度一半的位置(当然要分奇偶)。偶数的话就是K左右的两个数值相加除以2,奇数的话,也就是K所在的位置。那么对于两个有序数组的话,如果left1,left2(两个数组的左半部分)的元素个数相加等于K的时候,比较L1,L2找出最大的那个数值,那个数值就是K左边的值,同理,K右边的值就是R1,R2中最小的那个。若假设数组1的割点为C1,数组2的隔点为C2,那么当C1+C2 = K的时候,就可以进行比较。

4、比较的时候,如果L1 > R2的时候,就需要将C1往左移动(左边数值更小),将C2往右移动(右边数值更大),直到满足L1

同理,如果L2 > R1的时候,就需要将C2往左移动,C1往右边移动,直到满足L2

此处需要考虑越界的问题:

若当C1=0,移动到了最左边或者C2移动到了最右边,都还不满足L1 < R2的时候,说明数组1 > 数组2,此时中位数在数组2中,若当C2=0,移动到了最左边或者C1移动到了最右边,还不满足L2的时候,说明数组1 < 数组2,此时中位数在数组1中。

5、假设数组1长度为n,数组2长度为m,并且n

将数组长度进行虚处理,然后再把两个数组看作是数组A。

此时长度为(2m+1)+(2n+1)=2m+2n+2 此时从最中间隔开,k = (2m+2n+2)/2=(m+n+1),因为数组A是从下标0开始计数的,所以在程序中K实际等于(m+n).知道K值之后,C1,C2任意知道其中一个,便可以确定另外一个,此处对长度较小的数组进行二分,所以确定C1后再确定C2, C2 = k-C1=m+n-C1.此时我们将数组进行了虚拟的加#,所以要确定原来数组元素的真正位置,就包含一个映射关系。L1 =(C1-1)/2   L2=(C2-1)/2  R1=(C1/2) R2=(C2/2)
--------------------- 
作者:Jialuhu 
来源:CSDN 
原文:https://blog.csdn.net/qq_36573828/article/details/80752575 

 

代码如下:

class Solution {
public:
    double findMedianSortedArrays(vector& nums1, vector& nums2) {
        int n = nums1.size();
        int m = nums2.size();
        if(n > m)
        {
            return findMedianSortedArrays(nums2,nums1);
        }
        int L1,L2,R1,R2,c1,c2,lo = 0, hi = 2 * n;
        while(lo <= hi)
        {
            c1 = (lo + hi)/2;  
            c2 = m + n- c1;
            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)
                hi = c1 - 1;
            else if(L2 > R1)
                lo = c1 + 1;
            else
                break;
        }
        return (max(L1,L2)+ min(R1,R2))/2.0;
    }
};

码源: 

--------------------- 
作者:Vosky 
来源:CSDN 
原文:https://blog.csdn.net/hk2291976/article/details/51107778 
版权声明:本文为博主原创文章,转载请附上博文链接!

你可能感兴趣的:(算法)