堆有手写的,有STL自带的,我比较图省事,STL的已经满足这道题的需要,就直接用了。
这道题的思路是用2个堆,一个大根堆,一个小根堆,
我的处理方式是,每次读入一个数,只加入到一个堆里,以小根堆为例。
每次输出也是输出一个堆的堆顶元素,以大根堆为例,这样就要求维护小根堆堆顶大于大根堆堆顶,小根堆元素个数不得大于大根堆,只需要维护这两个性质就可以解决问题。
不过这道题十分的坑人啊,尤其是坑英文不好的人。
一开始连输入数据都没看懂。
一开始先输入一个数,代表数据组数,之后每组数据第一行是两个数,代表这是第几组数据和这组数据有多少个数,对于这个你需要输出的是两个数,第几组输出和这组数共有多少中位数。
更需要注意的是OUTPUT最后一行的描述,要求输出的每行最多10个数,并且结尾不能有多余空行(因为这个地方WA掉很多次,这种毛病一定要改啊啊啊啊啊,读题读题再读题)。
#include <cstdio>
#include <queue>
using namespace std;
int N, n, t;
priority_queue <int> big;
priority_queue < int, vector<int>, greater<int> > lit;
int main()
{
scanf("%d", &N);
while(N--){
while(big.size()) big.pop();
while(lit.size()) lit.pop();
int k = 0;
scanf("%d %d", &t, &n);
printf("%d %d\n", t, (n+1)/2);
for(int i = 1; i <= n; i++){
scanf("%d", &t);
lit.push(t);
while((big.size() && lit.size()) && big.top() > lit.top()){
int b = big.top(), l = lit.top();
big.pop(); lit.pop();
big.push(l), lit.push(b);
}
while(big.size() < lit.size()){
big.push(lit.top());
lit.pop();
}
if(i%2){
printf("%d", big.top()), k++;
if(k != 10) printf(" ");
else printf("\n"), k = 0;
}
}
if(N) printf("\n");
}
}