004-寻找两个正序数组的中位数-力扣

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

一：题目描述

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

示例3：

输入：nums1 = [0,0], nums2 = [0,0]
输出：0.00000

示例4：

输入：nums1 = [], nums2 = [1]
输出：1.00000

示例5：

输入：nums1 = [2], nums2 = []
输出：2.00000

题目来源：寻找两个正序数组中的中位数

二：思路及代码

(一)：思路

题目解析可分为两个部分，
1：将两个有序的数组合并成一个有序的数组。
2：在已合并的有序的数组中寻找中位数。

其中将两个有序的数组合并成一个有序的数组，其实就是归并排序，具体算法请见：排序算法合集
在已合并的数组中寻找中位数也很简单，若数组的长度为奇数，那么中位数就是array[array.length/2]，若数组的长度是偶数，那么中位数就是(array[array.length/2-1]+array[array.lenght/2])/2

(二)：代码

 public static double findMedianSortedArrays(int[] nums1, int[] nums2) {
        double array[] = new double[nums1.length+nums2.length];
        int i =0,j=0;
        int index = 0;
        while (i<nums1.length&&j<nums2.length){
            if (nums1[i]<nums2[j]){
                array[index++] = nums1[i++];
            }else {
                array[index++] = nums2[j++];
            }
        }
        while (i<nums1.length){
            array[index++] = nums1[i++];
        }
        while (j<nums2.length){
            array[index++] = nums2[j++];
        }
        for (int i1 = 0; i1 < array.length; i1++) {
            System.out.print(array[i1]+"\t");
        }

        //寻找中位数
        if (array.length%2 == 1){
            return array[array.length/2];
        }else {
            return (array[array.length/2]+array[array.length/2-1])/2;
        }
    }