题意
两个有序数组A,B,求两个数组合并后的中位数
思路
解法一:
使用归并排序的思路,遍历两个有序数组,然后当遍历数为我们要找到的中位数的位置的时候,即可停止。时间复杂度
O(n+m)
解法二:
根据中位数的定义可知,中位数左边的数小于等于中位数,中位数右边的数大于等于中位数,那么我门可以将左边的数看做一个集合
leftPart
,右边的数看做一个集合
rightPart
,当两个数组进行组合的时候,两个数组必定都会出现一个分割点,我们要找到这两个分割点,然后使得分割点左边的数都比中位数小,右边的数都比中位数大。
假设我们分割后的序列形式为
a[0],a[1],a[2],...,a[i - 1] | a[i], a[i + 1],...,a[n - 1]
b[0],b[1],b[2],...,b[j - 1] | b[j], b[j + 1],...,b[m - 1]
上面的
n,m
为数组A,B的长度,切割点分别为
i
,
如果数组A为较短的数组,且当前切割点为
i
,那么数组B的当前切割点也是确定了,通过等式
在我们寻找满足条件的
i
的时候,需要判断几种情况:
1.
这个时候说明我们切割点
i
太小了,需要增加,当
2.
这个时候说明我们切割点
i
太大了,需要减小,当
3.
这个时候说明我们找到了答案,但是需要处理一下边界情况(代码中很详细,不再赘述)
因为我们只是在一个数组中进行寻找切割点,且采用的是二分的方式,所以说时间复杂度为 O(log(min(n,m)))
结果
Your runtime beats 72.67% of cpp submissions.
代码
class Solution {
public:
double ans;
double findMedianSortedArrays(vector<int>& nums1, vector<int>& nums2) {
int n = nums1.size();
int m = nums2.size();
ans = -1;
if(n > m){
swap(n, m);
findNum(nums2, nums1, 0, n, n, m, ans);
}else{
findNum(nums1, nums2, 0, n, n, m, ans);
}
return ans;
}
private:
void findNum(vector<int>& nums1, vector<int>& nums2, int L, int R, int n, int m, double &ans){
if(ans >= 0) return ;
if(L > R) return ;
int i = (L + R) / 2;
int j = (n + m + 1) / 2 - i;
if(i < n && nums2[j - 1] > nums1[i]){
findNum(nums1, nums2, i + 1, R, n, m, ans);
} else if(i > 0 && nums1[i - 1] > nums2[j]){
findNum(nums1, nums2, L, i - 1, n, m, ans);
}else{
int maxL, minR;
if(i == 0) maxL = nums2[j - 1];
else if(j == 0) maxL = nums1[i - 1];
else maxL = max(nums1[i - 1], nums2[j - 1]);
if((n + m) % 2 != 0){
minR = -1;
}else{
if(i == n) minR = nums2[j];
else if(j == m) minR = nums1[i];
else minR = min(nums1[i], nums2[j]);
}
ans = (maxL + max(maxL, minR)) / 2.0;
}
}
};