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; }
【推荐】国内首个AI IDE,深度理解中文开发场景,立即下载体验Trae
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 【自荐】一款简洁、开源的在线白板工具 Drawnix
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· Docker 太简单,K8s 太复杂?w7panel 让容器管理更轻松!