4. 寻找两个正序数组的中位数 - 力扣(LeetCode)
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`。