4. 寻找两个正序数组的中位数(数组)

题目描述

给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。

算法的时间复杂度应该为 O(log (m+n)) 。

示例 1：

输入：nums1 = [1,3], nums2 = [2]
输出：2.00000
解释：合并数组 = [1,2,3] ，中位数 2
示例 2：

输入：nums1 = [1,2], nums2 = [3,4]
输出：2.50000
解释：合并数组 = [1,2,3,4] ，中位数 (2 + 3) / 2 = 2.5

提示：

nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-106 <= nums1[i], nums2[i] <= 106

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

方法一,重组排序

思路:将两个数组合并到另一个大数组中.其实可以再优化,在排序时采用更高效算法即可.

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int len1=nums1.length;
        int len2=nums2.length;
        int[] arr=new int[len1+len2];

        int k=0;
        int i,j;
        for( i=0,j=0;i<len1&&j<len2;){
            if(nums1[i]<=nums2[j]){
                arr[k]=nums1[i];
                i++;
            }else{
                arr[k]=nums2[j];
                j++;
            }
            k++;
        }
       while(i<len1){
           arr[k++]=nums1[i++];
       }
       while(j<len2){
           arr[k++]=nums2[j++];
       }

       if(arr.length%2==1){
           return arr[arr.length/2];
       }else{
       //这里值得注意,容易出错
          return (arr[(arr.length-1)/2]+arr[arr.length/2])/2.0;
       }
    }
}

方法二,调用系统函数

其实思路和方法一相同,只是系统已有数组拷贝函数和排序函数,故尝试调用,但似乎很耗时.

class Solution {
        public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int len1=nums1.length;
        int len2=nums2.length;
        int[] arr=new int[len1+len2];
        System.arraycopy(nums1,0,arr,0,len1);
        System.arraycopy(nums2,0,arr,len1,len2);
        Arrays.sort(arr);

        if(arr.length%2==1){
            return arr[arr.length/2];
        }else{
            return (arr[(arr.length-1)/2]+arr[arr.length/2])/2.0;
        }

    }
}