算法导论9.3-8习题解答(类似二分查找算法)

CLRS 9.3-8 ：

设X[1...n]和Y[1...n]为两个数组，每个都包含n个已排序好的数。给出一个求数组X和Y中所有2n个元素的中位数的、O(lgn)时间的算法。

算法思想：

该算法类似于二分查找算法

1.两个数组中小于median的个数为(n - 1)个，假设该median为数组a中的第k个，k为数组下标，那么在数组a中已经存在k个值小于median，那么在数组b中必然有(n - 1) - k = (n-k-1)个数小于median，如果b[n - k - 2] <= median <= b[n - k - 1]，那么median就找到了，如果median >= b[n - k - 1]，则搜索数组a中[k + 1, n - 1]中的元素，如果median <= b[n - k - 2]，则搜索数组a中[0, k - 1]中的元素.

2.依次类推，类似二分查找，不断重新设置low，high，middle

3.如果在数组a中没找到，则在数组b中找

#include <iostream>
using namespace std;
int search_median(int* a, int* b, int n, int low, int high);
int main()
{
constint LEN =6;
//两个已排序数组
int a[LEN] = {1, 2, 3, 4, 7, 9};
int b[LEN] = {2, 5, 6, 7, 10, 11};
int goal;
if((goal = search_median(a, b, LEN, 0, LEN -1)) ==-1)
goal = search_median(b, a, LEN, 0, LEN -1);
cout<<goal<<endl;
return 0;
}
int search_median(int* a, int* b, int n, int low, int high)
{
while(low <= high)
{
int middle = (low + high)/2;
//到达边界
if(middle == n -1&& a[middle] <= b[0])
return a[middle];
//非边界
else if(middle < n -1)
{
if(a[middle] <= b[n - middle -1] && a[middle] >= b[n - middle -2])
{
return a[middle];
}
else if(a[middle] >= b[n - middle -1])
{
high = middle -1;
}
else
{
low = middle +1;
}
}
}
return-1;
}