题目大意:
给两个非递减序列,求它们合并后中位数。
题目解析:
一开始用两个数组存储序列,然后计算求中位数,结果发现超出内存限制。改为存储第一个序列,第二序列边读边求,通过。
具体代码:
#include<iostream>
#include<algorithm>
#include<vector>
using namespace std;
#define MAXN 200005
int A[MAXN];
int B;
int main()
{
int m,n,middle;
int res;
scanf("%d",&m);
for(int i=0;i<m;i++){
scanf("%d",&A[i]);
}
scanf("%d",&n);
if((m+n)%2==0)
middle=(m+n)/2;
else
middle=(m+n)/2+1;
int k=0,i=0,j=0;
scanf("%d",&B);
while(i<m&&j<n&&k<middle){
if(A[i]<B)
res=A[i++];
else{
res=B;
scanf("%d",&B);
j++;
}
k++;
}
while(i<m&&k<middle){
res=A[i++];
k++;
}
while(j<m&&k<middle){
res=B;
scanf("%d",&B);
j++;
k++;
}
printf("%d",res);
return 0;
}