题目描述
给定两个大小为 m 和 n 的有序数组 nums1 和 nums2。
请你找出这两个有序数组的中位数,并且要求算法的时间复杂度为 O(log(m + n))。
你可以假设 nums1 和 nums2 不会同时为空。
示例 1:
nums1 = [1, 3]
nums2 = [2]
则中位数是 2.0
C++代码
class Solution {
public:
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int m = nums1.size();
int n = nums2.size();
if (m > n){
return findMedianSortedArrays(nums2, nums1);
}
int low = 0, high = m;
while(low <= high){
int i = (low + high) / 2;
int j = (m + n + 1) / 2 - i;
if (i > low && nums1[i-1] > nums2[j])
high = i - 1;
else if (i < high && nums2[j-1] > nums1[i])
low = i + 1;
else{
float maxLeft = 0;
if (i == 0) maxLeft = nums2[j-1];
else if (j == 0) maxLeft = nums1[i-1];
else maxLeft = max(nums1[i-1], nums2[j-1]);
if((m+n) & 1 == 1) return maxLeft;
float maxRight = 0;
if(j == n) maxRight = nums1[i];
else if (i == m) maxRight = nums2[j];
else maxRight = min(nums1[i], nums2[j]);
return (maxLeft + maxRight) / 2;
}
}
return 0;
}
};
题目思路
在一个集合中中位数起到一个划分的作用,它把这个集合划分为两个集合,其中一个集合中的任一元素小于另一集合中的任一元素。
在两个有序的数组中(nums1的长度为m,nums2的长度为n),我们分别在nums1的下标为i的元素前插入一个隔板,在nums2的下标为j的元素前插入隔板。
Left | Right |
---|---|
nums1[0] nums1[1] … nums[i-1] | nums1[i] nums1[i+1] … nums1[m] |
nums2[0] nums2[1] … nums2[j-1] | nums2[j] nums[j+1] … nums2[n] |
只要满足Left.len == Right.len(或者Left.len = Right.len + 1),并且Max(Left) <= Min(Right)我们就可以找到中位数。
即i + j = m - i + n - j (或m - i + n - j + 1) &&
max(nums1[i-1], nums2[j-1]) <= min(nums1[i], nums2[j])
由1式我们可以得出j = (m + n + 1) / 2 - i,因为是int型的计算,所以两个1式都可以满足。因为0 <= i <= m,而j = (m+n+1)/2 -i,为了保证j是一个合法的数,我们需要让m<=n。
由2式可得 nums1[i-1] <= nums2[j] && nums2[j-1] <= nums1[i]。
接下来我们可以按照以下步骤来进行二分搜索
- 令low = 0,high = m,在nums1中进行二分搜索
- 令i = (low + high) / 2, j = (m+n+1) / 2 - i
- 我们可以得到三种情况
- nums1[i-1] > nums2[j]
这种情况说明了i太大了需要缩小i,则令high = i - 1,继续二分 - nums2[j-1] > nums1[i]
这种情况说明了i太小了需要增大i,令low = i + 1,继续二分 - nums1[i-1] <= nums2[j] && nums2[j-1] <= nums1[i]
说明了我们找到了正确的i
- nums1[i-1] > nums2[j]
我们取左边的最大值为maxLeft,右边的最小值为minRight。当m+n是个奇数的时候,中位数就是maxLeft;m+n是个偶数的时候,中位数是(maxLeft + minRight) / 2,。接下来我们就需要考虑如何得到这个maxLeft和minRight。
-
当i == 0的时候,说明nums1的所有元素都在Right,那么maxLeft就是nums2中隔板前的那个元素,也就是nums2[j-1]
-
当j== 0的时候,说明nums2的所有元素都在Right,那么maxLeft就是nums1中隔板前的那个元素,也就是nums1[i-1]
-
不然的话,maxLeft = max(nums1[i-1], nums2[j-1])
-
当i == m的时候,说明nums1的所有元素都在Left,那么minRight就是nums2中隔板后的那个元素,也就是nums2[j]
-
当j == n的时候,说明nums2的所有元素都在Left,那么minRight就是nums1中隔板后的那个元素,也就是nums2[i]
-
不然的话,minRight = min(nums[i], nums[j])