寻找两个有序数组的中位数

问题描述

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

解题思路

把两个数组合并成一个新的有序数组，然后再寻找中位数。

代码

public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        
	 int len1 = nums1.length;
     int len2 = nums2.length;
     
     double[] merge = new double[len1 + len2];
     int i= 0 , i1 = 0, i2 = 0;
     while( i1 < len1 && i2 < len2){
         if(nums1[i1] > nums2[i2]) merge[i++] = nums2[i2++];
         else merge[i++] = nums1[i1++];
     }
     if(i1 < len1) for(int j = i1; j < len1; j++) merge[i++] = nums1[j];
     if(i2 < len2) for(int k = i2; k < len2; k++) merge[i++] = nums2[k];
         
     if((len1 + len2) % 2 == 0) return (merge[(len1 + len2)/2] + merge[(len1 + len2)/2 - 1])/2;
     else return  merge[(len1 + len2)/2];
        
    }