AcWing 106. 动态中位数
原题链接
中等
作者:
沙漠绿洲
,
2020-08-25 15:04:15
,
所有人可见
,
阅读 369
C++ 代码
#include <iostream>
#include <queue>
using namespace std;
int main()
{
int p;
cin >> p;
while(p --){
int b, n;
cin >> b >> n;
printf("%d %d\n", b, (n + 1) / 2);
priority_queue<int> q1; // 大顶堆
priority_queue<int, vector<int>, greater<int>> q2; // 小顶堆
int cnt = -1;
for(int i = 1; i <= n; ++ i)
{
int a;
scanf("%d", &a);
// 维护中位数
if(q1.size() <= q2.size()) { //加进大顶堆
q1.push(a);
while(!q2.empty() && q1.top() > q2.top()){
q2.push(q1.top());
q1.pop();
q1.push(q2.top());
q2.pop();
}
}else{ // 加进小顶堆
q2.push(a);
while(!q1.empty() && q1.top() > q2.top()){
q2.push(q1.top());
q1.pop();
q1.push(q2.top());
q2.pop();
}
}
//奇数时输出
if(i & 1){
if(++ cnt == 10) cnt = 0, printf("\n");
if(q1.size() > q2.size()) printf("%d ", q1.top());
else printf("%d ", q2.top());
}
}
printf("\n");
}
return 0;
}