3-3 两个有序序列的中位数 (20 分)
已知有两个非降序序列S1, S2, 求S1与S2归并成一个序列的低位中位数。有序序列A0,A1,⋯,AN−1的中位数指A(N−1)/2的值,即第⌊(N+1)/2⌋个数(A0为第1个数)。
输入格式:
输入分4行。第一行给出第一个序列的长度N1(0<N1≤2500000),随后是第一个序列的信息,即N1个非降序排列的整数。数字用空格间隔。随后是第二个序列的长度N2(0<N2≤2500000)和信息。因为测试数据只能10M,2.5*10的6次方规模,二分效果不明显,如果10的7次方数据规模,二分一般比归并快20ms左右。10的8次方快近200多ms。题目主要是输入花费大量的时间,如查找出现超时,尝试多次提交。
输出格式:
在一行中输出两个输入序列的并集序列的低位中位数。
输入样例:2
在这里给出一组输入。例如:
3
1 2 3
5
4 5 6 7 8
输出样例:
在这里给出相应的输出。例如:
4
#include<iostream>
using namespace std;
int main(){
int n1,n2;
cin>>n1;
int *a1 =(int*)malloc(sizeof(int)*(n1));
for(int i = 0;i < n1;i++){
scanf("%d",&a1[i]);
}
cin>>n2;
int *a2 =(int*)malloc(sizeof(int)*(n2));
for(int i = 0;i < n2;i++){
scanf("%d",&a2[i]);
}
int index=0,end=(n1+n2+1)/2,i=0,j=0;
int cur1=0,cur2=0,flag=0;
while(index!=end){
if(i<n1&&a1[i]<a2[j]){
cur1=i++;
flag=1;
}
else if(j<n2){
cur2=j++;
flag=2;
}
index++;
}
if(flag==1){
cout<<a1[cur1];
}
else{
cout<<a2[cur2];
}
return 0;
}