两个有序序列的中位数PTA

  • Post author:
  • Post category:其他


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

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



版权声明:本文为weixin_58073817原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。