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

题目

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

分析

将两个数组合并为一个数组，排序合并后的数组，最后返回排序的合并数组的中位数就可以

代码

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
    	//如果其中一个数组为空，直接返回另一个数组的中位数，这与数组元素个数有关
        if(nums1.length == 0)
			return ((nums2.length%2)==0)?(nums2[nums2.length/2]+nums2[nums2.length/2-1])/2.0:nums2[nums2.length/2];
		if(nums2.length == 0)
			return ((nums1.length%2)==0)?(nums1[nums1.length/2]+nums1[nums1.length/2-1])/2.0:nums1[nums1.length/2];
		//两个数组都不为空，将它们合并为一个数组，并进行排序
        int[] arr = new int[nums1.length + nums2.length];
        System.arraycopy(nums1, 0, arr, 0, nums1.length);
        System.arraycopy(nums2, 0, arr, nums1.length, nums2.length);
        Arrays.sort(arr);
        //返回合并后数组的中位数
        return ((arr.length%2)==0)?(arr[arr.length/2]+arr[arr.length/2-1])/2.0:arr[arr.length/2];
    }
}