[数据结构与算法]寻找两个正序数组的中位数(leetcode)

题目描述

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

示例

示例 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

解题思路

创建一个新数组，将两个数组放入新数组中，给新数组排序，找出中位数。
数组长度为偶数时中间两个数要求平均值。

注意：

返回值是double类型的，要注意 int转double 和 除法的先后顺序，不要损失精度。

代码

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int[] nums3=new int[nums1.length+nums2.length];
        int i,x,j,num1=nums1.length,num2=nums2.length,num3=nums3.length;
        double y;
        //将nums1写入新数组
        for(i=0;i<num1;i++){
            nums3[i]=nums1[i];
        }
        //将nums2写入新数组
        for(i=num1,j=0;j<num2;i++,j++){
            nums3[i]=nums2[j];
        }
        //数组排序
        Arrays.sort(nums3);
        //判断数组长度是单数还是双数，双数要取中间两个数求和去平均值，单数就是中间那个数。
        if(num3%2==0){
            x=num3/2;
            y= (double)(nums3[x]+nums3[x-1])/2;
        }else{
            x=num3/2;
            y=nums3[x];
        }
        return y;
    }
}