1029 Median (25)

最新推荐文章于 2019-02-07 23:51:33 发布

转载最新推荐文章于 2019-02-07 23:51:33 发布 · 500 阅读

pat 专栏收录该内容

37 篇文章

订阅专栏

本文介绍了一种高效算法来查找两个已排序数组的中位数，避免了传统方法的时间开销。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

自己做的运行超时

参考http://blog.youkuaiyun.com/sunbaigui/article/details/8657126

#include<vector>
#include<iostream>

int main()
{
int n1, n2;
while(scanf("%d",&n1)!=EOF)
{
std::vector<int> a, b;
for(int i = 0; i < n1; ++i)
{
int tmp;
scanf("%d",&tmp);
a.push_back(tmp);
}
scanf("%d",&n2);
for(int i = 0; i < n2; ++i)
{
int tmp;
scanf("%d",&tmp);
b.push_back(tmp);
}
//find median
int aIdx, bIdx, k;
aIdx = bIdx = 0;
int mIdx = (n1+n2-1)/2;
int cur;
for( k = -1; aIdx < n1 && bIdx < n2; )
{
if(a[aIdx] <= b[bIdx])
{
cur = a[aIdx];
++aIdx;
++k;
}
else
{
cur = b[bIdx];
++bIdx;
++k;
}
if(k == mIdx)
{
printf("%d\n",cur);
break;
}

}
if(k != mIdx && aIdx < n1)
printf("%d\n", a[aIdx+mIdx-k-1]);
if(k != mIdx && bIdx < n2)
printf("%d\n", b[bIdx+mIdx-k-1]);
}
return 0;
}