寻找两个正序数组的中位数-Leetcode04

如题：

给定两个大小为 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

代码：

/**
 * 给定两个大小为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。
 *
 * 请你找出这两个正序数组的中位数，并且要求算法的时间复杂度为 O(log(m + n))。
 *
 * 你可以假设 nums1 和 nums2 不会同时为空。
 */
public class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {

        int n=nums1.length;
        int m=nums2.length;
        int left=(n+m+1)/2;
        int right=(n+m+2)/2;
        return (getKth(nums1,0,n-1,nums2,0,m-1,left)+getKth(nums1,0,n-1,nums2,0,m-1,right))*0.5;
    }

    private int getKth(int []nums1,int start1,int end1,int[] nums2,int start2,int end2,int k){
        int len1=end1-start1+1;
        int len2=end2-start2+1;
        //让len1的长度小于len2，这样就能保证如果有数组空了，一定是len1
        if(len1>len2) {
            return getKth(nums2, start2, end2, nums1, start1, end1, k);
        }
        if(len1==0){
            return nums2[start2+k-1];
        }

        if(k==1){
            return Math.min(nums1[start1],nums2[start2]);
        }

        int i=start1+Math.min(len1,k/2)-1;
        int j=start2+Math.min(len2,k/2)-1;

        if(nums1[i]>nums2[j]){
            return getKth(nums1,start1,end1,nums2,j+1,end2,k-(j-start2+1));
        }else{
            return getKth(nums1,i+1,end1,nums2,start2,end2,k-(i-start1+1));
        }


    }




    /*public double findMedianSortedArrays(int[] nums1, int[] nums2) {

        int[] nums;
        int m= nums1.length;
        int n=nums2.length;
        nums=new int[m+n];
        if(m==0){
            if(n%2==0){
                return (nums2[n/2-1]+nums[n/2])/2.0;
            }else{
                return nums2[n/2];
            }
        }
        if(n==0){
            if(m%2==0){
                return (nums1[m/2-1]+nums1[m/2])/2.0;
            }else{
                return nums1[m/2];
            }
        }

        int count=0;
        int i=0,j=0;
        while(count!=(m+n)){
            //nums1数组先遍历完
            if(i==m){
                while(j!=n){
                    nums[count++]=nums1[j++];
                }
                break;
            }
            //nums2数组先遍历完
            if(j==n){
                while(i!=m){
                    nums[count++]=nums1[i++];
                }
                break;
            }

            //一般情况，对两个数组进行遍历，将数组元素放到第三个数组中
            if(nums1[i]<nums2[j]){
                nums[count++]=nums1[i++];
            }else{
                nums[count++]=nums2[j++];
            }
        }
        //如果数组元素个数为偶数，那么去中间两个数的平均值，如果是奇数，那么取中间的值
        if(count%2==0){
            return (nums[count/2-1]+nums[count/2])/2.0;
        }else{
            return nums[count/2];
        }
        *//**
         * 时间复杂度：遍历全部数组（m+n）
         * 空间复杂度：开辟了一个数组，保存合并后的两个数组O(m+n)
         *//*

    }*/
}

思路：转自Leetcode官网有才的程序员

链接https://leetcode-cn.com/problems/median-of-two-sorted-arrays/solution/xiang-xi-tong-su-de-si-lu-fen-xi-duo-jie-fa-by-w-2/

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/median-of-two-sorted-arrays
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。