力扣热门100题刷题笔记 - 4.寻找两个正序数组的中位数

题目链接：4.寻找两个正序数组的中位数

题目描述：

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

示例：

输入：nums1 = [1,3], nums2 = [2]
输出：2.00000
解释：合并数组 = [1,2,3] ，中位数 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

解题思路：

首先创建一个新的数组nums来保存合并后的结果。
使用两个指针mi和ni分别指向nums1和nums2的起始位置。
遍历nums1和nums2，比较当前指针位置上的元素大小，将较小的元素放入新数组nums中，并移动相应的指针。
如果其中一个数组已经全部合并到新数组中，将另一个数组中剩余的元素直接添加到新数组中。
最后，判断合并后数组的长度是否为奇数，如果是奇数，则返回中间位置的元素；如果是偶数，则返回中间两个元素的平均值。
时间复杂度： O(m + n)

代码：

public double findMedianSortedArrays(int[] nums1, int[] nums2) {

        int[] nums = mergeArrays(nums1, nums2);
        int len = nums.length;
        if ((len & 1) == 1) {
            return nums[len / 2];
        } else {
            return (nums[len / 2] + nums[len / 2 - 1]) / 2.0;
        }
    }

    private int[] mergeArrays(int[] nums1, int[] nums2) {
        int m = nums1.length;
        int n = nums2.length;
        int mi = 0, ni = 0;
        int[] ans = new int[m + n];
        int ai = 0;
        while (mi < m && ni < n) {
            if (nums1[mi] <= nums2[ni]) {
                ans[ai] = nums1[mi];
                ai++;
                mi++;
            } else {
                ans[ai] = nums2[ni];
                ai++;
                ni++;
            }
        }
        if (mi == m) {
            while (ni < n) {
                ans[ai] = nums2[ni];
                ni++;
                ai++;
            }
        } else {
            while (mi < m) {
                ans[ai] = nums1[mi];
                mi++;
                ai++;
            }
        }
        return ans;
    }