2(8) 两个有序序列的中位数 (20 分)

已知有两个等长的非降序序列S1, S2, 设计函数求S1与S2并集的中位数。有序序列A
0
​
,A
1
​
,⋯,A
N−1
​
的中位数指A
(N−1)/2
​
的值,即第⌊(N+1)/2⌋个数（A
0
​
为第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
注释：a[n]的中位数是A, b[n]数组的中位数是B
当n为奇数时，取较小中位数的后面部分和较大中位数前面的部分，包含两个中位数
当n为偶数时，不包含较小中位数。

#include <iostream>
using namespace std;

void find(int a[], int a_l, int a_r, int b[], int b_l, int b_r) {
  int mid1,mid2;
  int a_m, b_m;
if(a_r==a_l)
{
    mid2=a[a_r]<b[b_r]? a[a_r]:b[b_r];
    cout<<mid2;
    return;
}
  //如果查找数组包含两个数字，则直接找出中位数
  if (a_r - a_l == 1) {
    //第一个中位数是两个数组的开始元素中较大的
    mid1=a[a_l]<b[b_l]? b[b_l]:a[a_l];
    //第二个中位数是两个数组的结束元素中较小的
    mid2=a[a_r]<b[b_r]? a[a_r]:b[b_r];
    if (mid1 < mid2)
        cout  << mid1 << endl;
    else
        cout << mid2 << endl;
    return;
  }

  //第1个数组查找范围的中位数a_m。如果数组元素个数为偶数，有两个中位数，a_m取两个中位数中较大的。如果数组元素个数为奇数，则a_m取中间的数。
  a_m = (a_l + a_r + 1) / 2;
  //第2个数组查找范围的中位数b_m。如果数组元素个数为偶数，有两个中位数，b_m取两个中位数中较小的。如果数组元素个数为奇数，则b_m取中间的数。
  b_m = (b_l + b_r) / 2;

  //如果两个中位数相等，则找到退出
  if(a[a_m] == b[b_m])
  {
    cout << a[a_m] << endl;
    return;
  }

  if(a[a_m] < b[b_m])
    //第1个数组查找范围为右半部分，第2个数组查找范围为左半部分
    find(a, a_m, a_r, b, b_l, b_m);
  else
    //第1个数组查找范围为左半部分，第2个数组查找范围为右半部分
    find(a, a_l, a_m, b, b_m, b_r);
  return;

}

int main() {
  int n;
  cin >> n;
  int a[n];//数组1
  int b[n];//数组2

  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;
}