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

双指针算法： 

#include<iostream>
#include<algorithm>

using namespace std;

int n;
const int N = 1e5 + 10;
int a[N], b[N];

int main(){
    cin >> n;
    for(int i = 1; i <= n; i++) cin >> a[i];
    for(int i =  1; i <= n; i++) cin >> b[i];
    //双指针，一个指向a, 一个指向b
    int cnt = (n * 2 + 1) / 2;//注意是2 * n的长度
    int i = 1, j = 1, res = 0;//res记录答案
    while(i <= n && j <= n){
        if(a[i] < b[j]) res = a[i], i++;
        else res = b[j], j++;
        cnt--;
        if(cnt == 0){
            break;
        }
    }
    if(cnt == 0){
        cout << res;
        return 0;
    }
    //其中一个到头了，另一个还没有
    while(i <= n){
        res = a[i];
        i++;
        cnt--;
        if(cnt == 0) break;
    }
    while(j <= n){
        res = b[j];
        j++;
        cnt--;
        if(cnt == 0) break;
    }
    //output
    cout << res;
    
    return 0;
}