leetcode数组题目之4.寻找两个正序数组的中位数

题目描述：

 方法一：将两个数组合并，返回中位数即可。

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int m=nums1.size();
        int n=nums2.size();
        //一定要初始化，否则会出现空指针异常
        vector<int>nums(m+n,0);
        int i=0,j=0,k=0;
        while(i <= m-1 && j <= n-1)
        {
            if(nums1[i] < nums2[j])                
                nums[k++]=nums1[i++];
            else              
                nums[k++]=nums2[j++];
        }
        //处理nums2遍历完而nums1没有遍历完的情况
        while(i <= m-1)         
        {
            nums[n+i]=nums1[i];
            i++;
        }
        //处理nums1遍历完而nums2没有遍历完的情况
         while(j <= n-1)
        {
            nums[m+j]=nums2[j];
            j++;
        }
        //返回nums的中位数即为所求
        return (m+n)%2==0?(double)(nums[(m+n)/2-1]+nums[(m+n)/2])/2:(double)nums[(m+n)/2];
    }
};

方法二：

class Solution {
public:
    double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
        int m=nums1.size();
        int n=nums2.size();
        //确保m<n
        if(m>n){
            vector<int>tmp(nums1);
            nums1=nums2;
            nums2=tmp;
            m=nums1.size();
            n=nums2.size();
        }
        int iMin=0,iMax=m;
        while(iMin<=iMax){
            //nums1的分割线
            int i=(iMin+iMax)/2;
            //nums2的分割线
            int j=(m+n+1)/2-i;
            //保证左右部分长度相等，以及nums2[j-1]<=nums1[i],nums1[i-1]<=nums2[j]，主要控制i的增减
            if(i!=m&&j!=0&&nums1[i]<nums2[j-1])   iMin=i+1;
            else if(j!=n&&i!=0&&nums1[i-1]>nums2[j])   iMax=i-1;
            else{
                //分三种情况，讨论分割线左边的最大值
                int maxLeft=0;
                if(i==0)   maxLeft=nums2[j-1];
                else if(j==0)   maxLeft=nums1[i-1];
                else   maxLeft=max(nums1[i-1],nums2[j-1]);
                if((m+n)%2==1) return maxLeft;
                //分三种情况，讨论分割线右边的最小值
                int minRight=0;
                if(i==m)   minRight=nums2[j];
                else if(j==n)   minRight=nums1[i];
                else   minRight=min(nums1[i],nums2[j]);

                return (maxLeft+minRight)/2.0;
            }
        }
        return 0.0;    
    }
};