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

4. 寻找两个正序数组的中位数 - 力扣(LeetCode)

寻找两个正序数组的中位数_第1张图片

class Solution {
    public double findMedianSortedArrays(int[] nums1, int[] nums2) {
    int m = nums1.length;
    int n = nums2.length;

    int len = m + n;
    int left = -1, right = -1;
    int aStart = 0, bStart = 0;

    for (int i = 0; i <= len / 2; i++) {
        left = right;
        if (aStart < m && (bStart >= n || nums1[aStart] < nums2[bStart])) {
            right = nums1[aStart++];
        } else {
            right = nums2[bStart++];
        }
    }
    if ((len & 1) == 0)
        return (left + right) / 2.0;
    else
        return right;
    }
}

1. 初始化两个数组的起始位置:`aStart` 和 `bStart`,初始时均为 0。
2. 使用一个循环,循环的次数为 `(m + n) / 2 + 1`,这是因为要找到中位数,循环到达一半的位置即可。
3. 在循环中,用 `left` 记录上一轮的 `right`,然后根据当前 `aStart` 和 `bStart` 的位置判断下一个应该取的元素。
    - 如果 `aStart` 未越界且 `nums1[aStart] < nums2[bStart]`,则下一个元素为 `nums1[aStart]`,同时将 `aStart` 向后移动。
    - 否则,下一个元素为 `nums2[bStart]`,同时将 `bStart` 向后移动。
    - 更新 `right` 为当前选择的元素。
4. 循环结束后,根据数组长度的奇偶性,返回相应的结果:
    - 如果数组长度为偶数,则返回 `(left + right) / 2.0`。
    - 如果数组长度为奇数,则返回 `right`。

你可能感兴趣的:(算法,算法,java,leetcode,力扣)