AcWing 106. 动态中位数
题目链接:
正经题意:
依次读入一个整数序列,每当已经读入的整数个数为奇数时,输出已读入的整数构成的序列的中位数。
输入格式
第一行输入一个整数
P
\Rho
P,代表后面数据集的个数,接下来若干行输入各个数据集。
每个数据集的第一行首先输入一个代表数据集的编号的整数。
然后输入一个整数
M
\Mu
M,代表数据集中包含数据的个数,
M
\Mu
M 一定为奇数,数据之间用空格隔开。
数据集的剩余行由数据集的数据构成,每行包含 10个数据,最后一行数据量可能少于 10个,数据之间用空格隔开。
输出格式
对于每个数据集,第一行输出两个整数,分别代表数据集的编号以及输出中位数的个数(应为数据个数加一的二分之一),数据之间用空格隔开。数据集的剩余行由输出的中位数构成,每行包含 10 个数据,最后一行数据量可能少于 10个,数据之间用空格隔开。
输出中不应该存在空行。
简而言之:
输出每次数量为奇数时的中位数
思路:
利用对顶堆思路,一个存大的,一个存小的,每个存一半,奇数时中位数就是中间的,像沙漏
在输入数的时候不断维持两个堆的数量,让下面的最多多一个即可。
这题的思路非常妙,可惜能力不够,想不到这个堆顶堆的思想怎么扩展
代码:
#include <bits/stdc++.h>
using namespace std;
int main(){
int p,m;
scanf("%d",&p);
while(p--){
int cot;
scanf("%d",&cot);
scanf("%d",&m);
cout<<cot<<" "<<(m+1)/2<<endl;
int cnt = 0;
priority_queue<int> down;
priority_queue<int,vector<int>,greater<int> > up;
int x;
//代码跟着y总打的
for(int i = 1;i<=m;i++){
scanf("%d",&x);
if(down.empty() || x <= down.top()){
down.push(x);
}else{
up.push(x);
}
if(down.size() - 1 > up.size()){
up.push(down.top());down.pop();
}else if(down.size() < up.size()){
down.push(up.top());up.pop();
}
if(i%2==1){
cout<<down.top()<<" ";
cnt++;
if(cnt % 10 == 0){
cout<<endl;//输出要求
}
}
}
if(cnt%10!=0){
cout<<endl;//本题输出要求
}
}
}