题目

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

难度困难5435收藏分享切换为英文接收动态反馈

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

解法1：没有按照题目的O(log (m+n))来，直接暴力，先合并数组，再排序，找中位数

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        Integer[] arr = Stream.concat(Arrays.stream(nums1).boxed(), Arrays.stream(nums2).boxed()).toArray(Integer[]::new);
        Arrays.sort(arr);
        // System.out.println(Arrays.toString(arr));
        int len = arr.length;
        return len % 2 == 1 ? arr[len / 2] : ((arr[len / 2 - 1] + arr[len / 2]) / 2.0);
    }
}