两个有序序列的中位数PTA

已知有两个等长的非降序序列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

PS:找中位数有几种方法，一是新建一个长数组，通过排序，然后返回A[n-1/2]；二是通过课本上的递归算法（没写）；三是这题中的特殊情况，两个升序的数组找中位数，即先找下标，后返回数字即可；

#include <stdio.h>
void input(int arr[],int n){
    for(int i=0;i<n;i++){
        scanf("%d",&arr[i]);
    }
}
int main(){
    int n;
    scanf("%d",&n);
    int a[n],b[n];
    input(a,n);
    input(b,n);
    int r=0,s=0;
    while(r+s<n-1){//当r+s==下标n-1时，为中位数；
        if(a[r]<=b[s]){
            r++;
        }else{
            s++;
        }
    }
    int mid=a[r]<b[s]?a[r]:b[s];
    printf("%d",mid);
    return 0;
}

值得一提的是，一般计算中位数会取中间值，这题直接舍去一点精度换速度，直接取小的值即可。