2-6 两个有序序列的中位数 (20 分)
已知有两个等长的非降序序列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 <stdio.h>
#define MAXSIZE 100000
int main(){
int N=0;
scanf("%d",&N);
int *p,*q;
p=(int *)malloc(N*sizeof(int));
q=(int *)malloc(N*sizeof(int));
int i=0,j=0;
for(i=0;i<N;i++){
scanf("%d",&p[i]);
}
for(j=0;j<N;j++){
scanf("%d",&q[j]);
}
int a[2*N];
int size=0;
i=0;j=0;
int t=0;
for(t=0;t<2*N;t++){
if(p[i]>=q[j]) {
a[size]=q[j];j++;
}else {a[size]=p[i];i++;}
size++;
}
printf("%d",a[size/2-1]);
}