PTA 1029 Median

题目描述

分析：

解法1：双指针，从两个序列开始处移动，直到到达中位数。

#include<cstdio>
int a[1000005],b[1000005];
int main(){
	int n,m;
	scanf("%d",&n);
	for(int i=0;i<n;i++)
	scanf("%d",&a[i]);
	scanf("%d",&m);
	for(int i=0;i<m;i++)
	scanf("%d",&b[i]);
	a[n]=b[m]=0x7fffffff;
	int mid=(n+m-1)/2;
	int i=0,j=0,count=0;
	while(count<mid){
		if(a[i]<b[j]) i++;
		else j++;
		count++;
	}
	if(a[i]<b[j])
	printf("%d",a[i]);
	else
	printf("%d",b[j]);
	return 0;
}

解法2：合并两个序列，排序输出中位数

#include<cstdio>
#include<algorithm>
using namespace std;
int a[2000005],b[1000005];
int main(){
	int n,m;
	scanf("%d",&n);
	for(int i=0;i<n;i++)
	scanf("%d",&a[i]);
	scanf("%d",&m);
	for(int i=n;i<m+n;i++)
	scanf("%d",&a[i]);
	int mid=(n+m-1)/2;
	sort(a,a+m+n);
	printf("%d",a[mid]);
	return 0;
}