LeetCode 练习——4. 寻找两个正序数组的中位数

1.题目描述

寻找两个正序数组的中位数
给定两个大小分别为 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

2.思路

2.1 代码

使用归并排序的思想合并两个数组，再求出中位数。
在合并后的数组求中位数时，数组长度为奇数，则直接使用中间那一位数字作为结果；如果长度为偶数，则使用中间两位之和除以2作为结果

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
        double[] mergeArray = merge(nums1, nums2);
        int mid = mergeArray.length / 2;
        return mergeArray.length % 2 == 0 ? (mergeArray[mid - 1] + mergeArray[mid]) / 2 : mergeArray[mid];
    }

    public double[] merge(int[] nums1, int[] nums2) {
        double[] mergeArray = new double[nums1.length + nums2.length];
        int ptr1 = 0;
        int ptr2 = 0;
        int i = 0;
        while (ptr1 < nums1.length && ptr2 < nums2.length) {
            mergeArray[i++] = nums1[ptr1] <= nums2[ptr2] ? nums1[ptr1++] : nums2[ptr2++];
        }
        while (ptr1 < nums1.length) {
            mergeArray[i++] = nums1[ptr1++];
        }
        while (ptr2 < nums2.length) {
            mergeArray[i++] = nums2[ptr2++];
        }
        return mergeArray;
    }
}

2.2 测试结果

通过测试

3.总结

• 使用归并排序的思想合并数组
• 得到合并之后的数组求中位数。合并数组长度为奇数，返回中间值；长度为偶数，返回中间两位数的平均值