题目链接:题目链接
题意:两个排好序的数组,找到中位数,如果是奇数好办,如果是偶数找到最中间的两个求平均值。
这一题的本质其实就是第mid小的数。
这一题看到一种好的方法,我们令k=mid,对于两个数组,分别取前k/2数,如果A[k/2-1]比B[k/2-1]大,那么说明B前 k/2数肯定在k小数中,A的则不一定,则下面需要从A,B+k/2再次去寻找。。。反之对于B大的情况也是一样的,如果相 等,那就是这两个数(相等)了,随便返回一个都OK。(Ps:解释一些为什么是k/2-1,因为k是从1开始取,数组下标从0开 始)。
代码如下:
class Solution {
public:
// 参数K:从1开始
//
double dooo(int A[], int m, int B[], int n, int k){
// 为了方便处理,默认m比n小
//
if (m > n)
return dooo(B, n, A, m, k);
// 如果数组A没有元素了,那么返回B中k-1就OK
//
if (!m)
return B[k-1];
if (k==1)
return min(A[0], B[0]);
// A中前面k/2数,如果比m大,那么取m
// 对于b,那么和a_idx和是k
//
int a_idx = min(k/2, m);
int b_idx = k - a_idx;
// 如果A比B值大,说明B中前b_idx数都在K小数中
// 反之一样
//
if (A[a_idx-1] > B[b_idx-1])
// 如果B前b_idx数都在前k小数中,那么下面从A和B+b_idx再次进行递归
//
return dooo(A, m, B+b_idx, n-b_idx, k-b_idx);
else if (A[a_idx-1] < B[b_idx-1])
return dooo(A+a_idx, m-a_idx, B, n, k-a_idx);
else
return A[a_idx-1];
}
double findMedianSortedArrays(int A[], int m, int B[], int n) {
// 如果是偶数,需要返回中间两个数,获得平均数
// 注意第三个参数k从1开始,所以需要+1
//
if ((n + m) % 2 == 0)
return (dooo(A, m, B, n, (m+n)/2) + dooo(A, m, B, n, (m+n)/2+1)) / 2.0;
else
// 否则中间数就OK
return dooo(A, m, B, n, (m+n)/2+1);
}
};