LeetCode 系列 第二题 寻找两个正序数组的中位数

LeetCode 系列 第二题 寻找两个正序数组的中位数

题目要求

给定两个大小分别为 m 和 n 的正序（从小到大）数组 nums1 和 nums2。请你找出并返回这两个正序数组的 中位数 。
算法的时间复杂度应该为 O(log (m+n)) 。
示例 1：
输入：nums1 = [1,3], nums2 = [2]
输出：2.00000
解释：合并数组 = [1,2,3] ，中位数 2

题目解析

要找两个有序数组的中位数，可以利用归并排序的思想将两个数组合并成一个有序数组，在合并过程中找到中位数即可。根据题目要求算法的时间复杂度应该为 O(log (m+n))，所以不能使用简单的合并排序的方法。
以下是一种满足时间复杂度要求的解法：

首先判断两个数组的总长度（m + n）的奇偶性。如果总长度为奇数，中位数位置为 (m + n) / 2 + 1，如果总长度为偶数，则中位数位置为 (m + n) / 2 和 (m + n) / 2 + 1。
定义变量 count 记录当前合并的元素个数，初始化为 0。
定义变量 prev 和 curr 分别表示前一个和当前的合并元素的值，初始化为 0。
使用两个指针 p1 和 p2 分别指向数组 nums1 和 nums2 的起始位置。
进行循环，循环条件为 count < (m + n) / 2 + 1：
对两个数组的当前元素进行比较，选择较小的元素进行合并。
更新 prev 和 curr 的值。
指针 p1 和 p2 向后移动。
每次合并后，将 count 的值加一。
如果总长度为奇数，则中位数就是 curr 的值；如果总长度为偶数，则中位数为 (prev + curr) / 2.0。
返回中位数的值。

代码示例

// 

    public static double findMedianSortedArrays(int[] nums1, int[] nums2) {
        int m = nums1.length;
        int n = nums2.length;
        int totalLength = m + n;
        int midIndex1 = (totalLength + 1) / 2;
        int midIndex2 = (totalLength + 2) / 2;
        return (findKthElement(nums1, 0, nums2, 0, midIndex1) + findKthElement(nums1, 0, nums2, 0, midIndex2)) / 2.0;
    }

    private static int findKthElement(int[] nums1, int start1, int[] nums2, int start2, int k) {
        if (start1 >= nums1.length) {
            return nums2[start2 + k - 1];
        }
        if (start2 >= nums2.length) {
            return nums1[start1 + k - 1];
        }
        if (k == 1) {
            return Math.min(nums1[start1], nums2[start2]);
        }
        int midIndex1 = start1 + k / 2 - 1;
        int midIndex2 = start2 + k / 2 - 1;
        int midValue1 = midIndex1 < nums1.length ? nums1[midIndex1] : Integer.MAX_VALUE;
        int midValue2 = midIndex2 < nums2.length ? nums2[midIndex2] : Integer.MAX_VALUE;
        if (midValue1 <= midValue2) {
            return findKthElement(nums1, midIndex1 + 1, nums2, start2, k - k / 2);
        } else {
            return findKthElement(nums1, start1, nums2, midIndex2 + 1, k - k / 2);
        }
    }