PTA 两个有序序列的中位数

344 阅读1分钟

已知有两个等长的非降序序列S1, S2, 设计函数求S1与S2并集的中位数。有序序列A0,A1,,AN1A_0,A_1,⋯,A_{N−1}的中位数指A(N1)/2A_{(N−1)/2}的值,即第⌊(N+1)/2⌋个数(A0A_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

代码:

# include <iostream>
using namespace std;
const int N =100005;
int main()
{
    int  n;
    cin >> n;
    int a[N],b[N];
    for(int i=0;i<n;i++)
        cin >> a[i];
    for(int i=0; i<n; i++)
        cin >> b[i];
    int i,j;
    for(i=0,j=0;i+j<(n+n-1)/2;)
    {
        if(a[i]>b[j])
            j++;
        else
            i++;
    }
    if(a[i]<b[j])
        cout << a[i] << endl;
    else
        cout << b[j] << endl;
}

提交结果: