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

又到了每日一题的时间，今天我们来做第4题，leetcode对这题的标记是困难，但是做起来好像还是很简单，可能因为还是前几题的原因吧。那么我们先来看看题目：

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

第一步，先来看看解题步骤（依旧附上我丑丑的字）：

（1）创建一个新数组存放两个数组的元素，长度是两个数组的长度之和。

（2）将两个数组的元素依次复制到新数组中。

（3）将新数组中的元素进行排序，建议使用归并排序，快速排序，希尔排序，堆排序这些时间复杂度更小的排序方法，我是直接调用了sort方法，算是优化后的快速排序，当然大家也可以自己写排序算法。

（4）求中位数

下面看代码：

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);
        int index = (len1 + len2)/2;
        if((len1 + len2)%2 == 0)
        {
            double num1 = arr[index-1];
            double num2 = arr[index];
            return (num1 + num2)/2;             
        }
        else
        {           
            return arr[index];
        }
    }
}

 

提交结果：

执行用时 :3 ms, 在所有 Java 提交中击败了84.30%的用户

内存消耗 :41.1 MB, 在所有 Java 提交中击败了98.36%的用户

 

欢迎评论留言，最迟一天内回复噢。