题目大意:给出两个分别不超过2*10^5的升序序列,要求找出它们合并后的序列的中位数。内存限制1.5MB。
因为内存限制太小,不能同时存储两个序列,只能存储第一个序列,然后再读入第二个序列,在读入的同时进行合并操作,直到找到应当是整个序列的中位数的位置的数位置。
注意点:①若序列的长度为N,则中位数的位置是(N+1)/2,从下标为1开始计。②若第一个序列的当前位置小于正在读入的数据,需要遍历序列1直到大于当前读入的数据。③如果读完了第2个序列,还是没有找到中位数,说明中位数在序列1的当前位置的后面,需要额外处理。
两组自测的数据:
① 5 9 10 15 16 17 // 2 3 4 ② 5 9 10 15 16 17 // 2 3 12
AC代码如下:
#include <vector>
#include <cstdio>
using namespace std;
int main()
{
int N;
scanf("%d", &N);
vector<int> v(N);
for (int i = 0; i < N; ++i)
scanf("%d", &v[i]);
int K;
scanf("%d", &K);
int midIndex = (N + K + 1) / 2;
int index = 0, i = 0;
for (int j = 0; j < K; ++j)
{
int tmp;
scanf("%d", &tmp);
while(i < N && v[i] <= tmp)
{
if(++index == midIndex)
{
printf("%d", v[i]);
break;
}
i++;
}
if(++index == midIndex)
{
printf("%d", tmp);
break;
}
}
if(index < midIndex)
printf("%d", v[i + midIndex - index - 1]);
return 0;
}