PTA刷题第16题
7-1 两个有序序列的中位数 (20分)
已知有两个等长的非降序序列S1, S2, 设计函数求S1与S2并集的中位数。有序序列A0 ,A1 ,⋯,AN−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
注意这里很坑的是:两个序列的并集并不需要考虑去重的因素!!!
思路:分别把两个序列读入到两个数组中,之后将两个数组中的数进行比较,小的读进新的数组,最后如果两个数组中还有一个数组还有剩余,那么就让其直接放入新的数组后面即可。代码如下;
#include<stdio.h>
int main(){
int n;
scanf("%d",&n);
int a1[n];
int a2[n];
int a3[2*n];
for(int i=0;i<n;i++){
scanf("%d",&a1[i]);
}
for(int i=0;i<n;i++){
scanf("%d",&a2[i]);
}
int a,b,c;
a=0,b=0,c=0;//这是三个数组的指针
while(a<n && b<n){//比较a1指针数和a2指针数的大小关系,小的进入a3
if(a1[a]<a2[b]){
a3[c++]=a1[a++];
}
else{
a3[c++]=a2[b++];
}
}
while(a<n){//如果a1合并之后还有剩余,则进入a3
a3[c++]=a1[a++];
}
while(b<n){
a3[c++]=a2[b++];
}
printf("%d",a3[(c-1)/2]);
}