acwing106

如果说对静态的数据找中位数,那么用快排或直接用partion方法来做即可。但当前题目要求的是动态的读取数据来输出中位数,即输出前1、3、5、7...个数据的中位数,那么就需要使用对顶堆来处理。
对顶堆使用两个堆,上面的堆up是小根堆,下面的堆down是大根堆,且必须满足小根堆的top要大于大根堆的top。每读入一个新的数据i,先判断数据i与大根堆top值mid的大小关系,如果i更大,就将其加入小根堆up,否则加入大根堆down。然后是关键的平衡步骤:根据当前已处理的数据个数,进行操作:如果两个堆中共有奇数个元素,则down必须比up大一个元素,通过top与pop操作实现;如果共有偶数个元素,则up和down的元素个数必须相同,通过top与pop操作实现。经过这样初始化和操作后的堆在输入奇数个元素时,中位数为down的top;在输入偶数个元素时,中位数为down的top和up的top。
C++的STL已有优先队列容器,其内部实现正是堆,只需要指定其是小根堆还是大根堆即可使用。
代码如下:

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<string>
using namespace std;
int P, id,M;
int main(void) {
//加速IO
ios::sync_with_stdio(false);
cin >> P;
int c = 0;
while (++c<=P) {
priority_queue<int, vector<int>, greater<int>> up; //小根堆
priority_queue<int, vector<int>, less<int>> down; //大根堆
cin >>id>> M;
cout << id << " " << (M + 1) / 2 << endl;
int cnt = 0; //记录已经输出的结果数
for (int i = 1; i <= M; i++) {
int x; cin >> x;
if (down.size() == 0) down.push(x);
else {
if (x > down.top())
up.push(x);
else
down.push(x);
}
if (i % 2 == 0) {
//平衡两个堆的大小
if (down.size() != up.size()) {
up.push(down.top());
down.pop();
}
}
else {
if (down.size() < up.size()) {
down.push(up.top());
up.pop();
}
//输出当前中位数
cout << down.top();
cnt++;
if (cnt % 10 == 0 || cnt==(M+1)/2)
cout << endl;
else
cout << " ";
}
}
}
return 0;
}
posted @   带带绝缘体  阅读(21)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 【自荐】一款简洁、开源的在线白板工具 Drawnix
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· Docker 太简单,K8s 太复杂?w7panel 让容器管理更轻松!
点击右上角即可分享
微信分享提示