AcWing 106 动态中位数 (对顶堆)

https://www.acwing.com/problem/content/108/

维护一个大根堆,一个小根堆,设当前序列长度为\(M\)
当前序列从小到大排名\(1~M/2\)的整数存在大根堆
排名\(M/2+1~M\)的整数存在小根堆,
如果插入后某一堆元素过多,就把该堆堆顶取出来插入令一个堆,
这样序列的中位数就是小根堆的堆顶

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<iostream>
#include<cmath>
#include<stack>
#include<queue>
using namespace std;
typedef long long ll;

const int maxn = 100010;

int T, n, Case, cnt;
int ans[maxn];
priority_queue<int> qma;
priority_queue<int, vector<int>, greater<int> > qmi;

ll read(){ ll s=0,f=1; char ch=getchar(); while(ch<'0' || ch>'9'){ if(ch=='-') f=-1; ch=getchar(); } while(ch>='0' && ch<='9'){ s=s*10+ch-'0'; ch=getchar(); } return s*f; }

int main(){
	T = read();
	while(T--){
		while(!qma.empty()) qma.pop();
		while(!qmi.empty()) qmi.pop();
		
		cnt = 0;
		Case = read(), n = read();
		int x;
		for(int i=1;i<=n;++i){
			x = read();
			if(i==1) qmi.push(x);
			else{
				if(x < qmi.top()) qma.push(x);
				else qmi.push(x);
			} 
			
			while(qma.size() > (i/2)){
				int tmp = qma.top();
				qmi.push(tmp);
				qma.pop();
			}
			int S;
			if(i%2 == 1) S = i/2 + 1;
			else S = i/2;
			while(qmi.size() > S){
				int tmp = qmi.top();
				qma.push(tmp);
				qmi.pop();
			}
			
			if(i%2 == 1) ans[++cnt] = qmi.top();
		}
		
		printf("%d %d\n",Case,cnt);
		int tot = 0;
		for(int i=1;i<=cnt;++i){
			if(tot == 10){ printf("\n"); tot = 0; }
			printf("%d ",ans[i]);
			++tot;
		}printf("\n"); 
	}
	return 0;
}
posted @ 2020-11-04 21:37  Tartarus_li  阅读(71)  评论(0编辑  收藏  举报