仅做学习笔记之用。
题目:
给定两个大小分别为 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
提示:
nums1.length == m
nums2.length == n
0 <= m <= 1000
0 <= n <= 1000
1 <= m + n <= 2000
-106 <= nums1[i], nums2[i] <= 106
其实就是二分查找嘛。如果没了解过二分查找的可以看一下这篇文章
这段代码首先确保较短的数组是nums1
,然后使用二分查找在nums1
上寻找合适的分割点。在找到合适的分割点后,我们可以计算出中位数。如果两个数组的总长度是奇数,中位数就是左边部分的最大值;如果是偶数,中位数就是左边部分的最大值和右边部分的最小值的平均值。
代码如下:
class Solution {
public static double findMedianSortedArrays(int[] nums1, int[] nums2) {
int m = nums1.length;
int n = nums2.length;
// 确保nums1是较短的数组
if (m > n) {
int[] temp = nums1;
nums1 = nums2;
nums2 = temp;
int tempLen = m;
m = n;
n = tempLen;
}
int Min = 0, Max = m, halfLen = (m + n + 1) / 2;
while (Min <= Max) {
int i = (Min + Max) / 2;
int j = halfLen - i;
if (i < m && nums2[j - 1] > nums1[i]) {
Min = i + 1; // i太小
} else if (i > 0 && nums1[i - 1] > nums2[j]) {
Max = i - 1; // i太大
} else {
// 找到合适的i
int maxLeft = (i == 0) ? nums2[j - 1] :
(j == 0) ? nums1[i - 1] : Math.max(nums1[i - 1], nums2[j - 1]);
// 如果总长度是奇数,直接返回中位数
if ((m + n) % 2 == 1) {
return maxLeft;
}
int minRight = (i == m) ? nums2[j] :
(j == n) ? nums1[i] : Math.min(nums1[i], nums2[j]);
// 如果总长度是偶数,返回中间两个数的平均值
return (maxLeft + minRight) / 2.0;
}
}
return 0;
}
}
(算法大佬英雄哪里出来曾曰:过——啦)
解决。