leetcode 04:寻找两个有序数组的中位数

• 给定两个大小为 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，思路是两个有序链表拼接的思路，两个数组遍历

class Solution {
    public static  double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int i=0;int j=0;int k=0;int l=0;
        if((nums1.length+nums2.length)%2==0)
        {
            k=(nums1.length+nums2.length)/2;
            while(i<nums1.length&&j<nums2.length)
            {
                if(nums1[i]<nums2[j])
                {
                    l=nums1[i];
                    i++;
                    k--;
                }
                else
                {
                    l=nums2[j];
                    j++;
                    k--;
                }
                if(k==0)
                {
                    break;
                }
            }
            if(k==0)
            {
                if(i!=nums1.length&&j!=nums2.length)
                {
                    if(nums1[i]<nums2[j])
                    {
                        return (l+nums1[i])/2.0;
                    }
                    else {
                        return (l+nums2[j])/2.0;
                    }
                }else if(i!=nums1.length)
                {
                    return (l+nums1[i])/2.0;
                }else {
                    return (l+nums2[j])/2.0;
                }
            }else
            {
                if(i!=nums1.length)
                {
                    return (nums1[i+k-1]+nums1[i+k])/2.0;
                }else
                {
                    return (nums2[j+k-1]+nums2[j+k])/2.0;
                }
            }

        }else{
            k=(nums1.length+nums2.length+1)/2;
            while(i<nums1.length&&j<nums2.length)
            {
                if(nums1[i]<nums2[j])
                {
                    l=nums1[i];
                    i++;
                    k--;
                }
                else
                {
                    l=nums2[j];
                    j++;
                    k--;
                }
                if(k==0)
                {
                    return Double.valueOf(l);
                }
            }
            if(i!=nums1.length)
            {
                return Double.valueOf(nums1[i+k-1]);
            }else
            {
                return Double.valueOf(nums2[j+k-1]);
            }

        }
    }
}

• 暴力破解的优化，松散代码结构

class Solution {
    public static  double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int i=0;int j=0;int k=0;int l=0;
        k = (nums1.length+nums2.length+1)/2;
        while(i<nums1.length&&j<nums2.length)
        {
            if(nums1[i]<nums2[j])
            {
                l=nums1[i];
                i++;
                k--;
            }
            else
            {
                l=nums2[j];
                j++;
                k--;
            }
            if(k==0)
            {
                if((nums1.length+nums2.length)%2!=0)
                    return Double.valueOf(l);
                else{
                    if(i!=nums1.length&&j!=nums2.length)
                    {
                        if(nums1[i]<nums2[j])
                        {
                            return (l+nums1[i])/2.0;
                        }
                        else {
                            return (l+nums2[j])/2.0;
                        }
                    }else if(i!=nums1.length)
                    {
                        return (l+nums1[i])/2.0;
                    }else {
                        return (l+nums2[j])/2.0;
                    }                    
                }
            }
        }
        if((nums1.length+nums2.length)%2!=0)
        {
                if(i!=nums1.length)
                {
                    return Double.valueOf(nums1[i+k-1]);
                }else
                {
                    return Double.valueOf(nums2[j+k-1]);
                }
        }else{
                if(i!=nums1.length)
                {
                    return (nums1[i+k-1]+nums1[i+k])/2.0;
                }else
                {
                    return (nums2[j+k-1]+nums2[j+k])/2.0;
                }
        }

    }
}

• 官方标准解,就是二分搜索的方法

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        if(nums1.length > nums2.length){
            //确报nums1始终比nums2小
            int [] temp = nums1;
            nums1 = nums2;
            nums2 = temp;
        }
        int length1 = nums1.length;
        int length2 = nums2.length;
        int left = 0;
        int right = length1;
        int mid = (length1+length2+1)/2;//不管奇数还是偶数，都是中位数的位置之一
        while (left<=right){
            int i = (left+right)/2;
            int j = mid-i;
            //上面两句话是移动后的中位数位置，这里的中位数是代表总体的中位数的预测位置
            if(i<right&&nums2[j-1]>nums1[i]){
                left = i+1;//向右扩张
            }else if(i>left&&nums1[i-1]>nums2[j]){
                right = i-1;//向左扩张
            }else {
                int maxleft;
                if(i==0){
                    maxleft = nums2[j-1];
                }else if(j==0){
                    maxleft = nums1[i-1];
                }else {
                    maxleft = Math.max(nums1[i-1],nums2[j-1] );
                }
                if((length1+length2)%2==1)
                    return maxleft;
                else {
                    int minRight = 0;
                    if(i==length1)
                        minRight = nums2[j];
                    else if (j==length2)
                        minRight = nums1[i];
                    else
                        minRight = Math.min(nums1[i],nums2[j]);
                    return (maxleft+minRight)/2.0;
                }
            }
        }
        return 0.0;

    }
}