PAT——1029 Median (25 分)

题目大意：

给两个非递减序列，求它们合并后中位数。

题目解析：

一开始用两个数组存储序列，然后计算求中位数，结果发现超出内存限制。改为存储第一个序列，第二序列边读边求，通过。

具体代码：

#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;
}