7-3 两个有序序列的中位数 (20分)
已知有两个等长的非降序序列S1, S2, 设计函数求S1与S2并集的中位数。有序序列A0,A1,⋯,AN−1的中位数指A(N−1)/2的值,即第⌊(N+1)/2⌋个(A0为第1个数)。
输入格式:
输入分三行。第一行给出序列的公共长度N(0<N≤100000),随后每行输入一个序列的信息,即N个非降序排列的整数。数字用空格间隔。
输出格式:
在一行中输出两个输入序列的并集序列的中位数。
输入样例1:
5
1 3 5 7 9
2 3 4 5 6
输出样例1:
4
输入样例2:
6
-100 -10 1 1 1 1
-50 0 2 3 4 5
输出样例2:
1
#include <iostream>
using namespace std;
void find(int a[], int a_l, int a_r, int b[], int b_l, int b_r) {
int m1, m2;
//如果查找数组包含两个数字,则直接找出中位数 4 6 /2 3 m1取4,m2取3
if (a_r - a_l == 1) {
//第一个中位数是两个数组的开始元素中较大的
m1=a[a_l]<b[b_l]? b[b_l]:a[a_l];
//第二个中位数是两个数组的结束元素中较小的
m2=a[a_r]<b[b_r]? a[a_r]:b[b_r];
if (m1 < m2)
cout << m1 << endl;
else
cout << m2 << endl;
return;
}
//如果数组元素个数为奇数,则m都取中间的数。
//如果数组元素个数为偶数,有两个中位数,m1取两个中位数中较大的。
m1 = (a_l + a_r + 1) / 2;
//如果数组元素个数为偶数,有两个中位数,m2取两个中位数中较小的。
m2 = (b_l + b_r) / 2;
//如果ab两个中位数相等,则a[m1]=a[m2]就是中位数
if(a[m1] == b[m2])
{
cout << a[m1] << endl;
return;
}
if(a[m1] < b[m2])
//第1个数组查找范围为右半部分,第2个数组查找范围为左半部分
find(a, m1, a_r, b, b_l, m2);
else
//第1个数组查找范围为左半部分,第2个数组查找范围为右半部分
find(a, a_l, m1, b, m2, b_r);
return;
}
int main() {
int n;
cin >> n;
int a[n];
int b[n];
for (int i = 0; i < n; i++)
cin >> a[i];
for (int i = 0; i < n; i++)
cin >> b[i];
find(a, 0, n - 1, b, 0, n - 1);
return 0;
}