BZOJ1112: [POI2008]砖块Klo（洛谷P3466）

平衡树

BZOJ题目传送门
洛谷题目传送门

动态维护中位数，平衡树上一发就好了。

代码：

#include<cctype>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#define N 500005
#define F inline
using namespace std;
typedef long long LL;
struct tree{ LL x,s; int r,sz,p,to[2]; }t[N];
int n,m,k,rt,h[N],mid,ls; LL ans=1e18,ans1,ans2;
F char readc(){
	static char buf[100000],*l=buf,*r=buf;
	if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
	return l==r?EOF:*l++;
}
F int _read(){
	int x=0; char ch=readc();
	while (!isdigit(ch)) ch=readc();
	while (isdigit(ch)) x=(x<<3)+(x<<1)+(ch^48),ch=readc();
	return x;
}
F void pshp(int x){
	t[x].sz=t[t[x].to[0]].sz+t[t[x].to[1]].sz+t[x].p;
	t[x].s=t[x].x*t[x].p+t[t[x].to[0]].s+t[t[x].to[1]].s;
}
F void rtt(int &x,int l){
	int s=t[x].to[l]; t[x].to[l]=t[s].to[l^1];
	t[s].to[l^1]=x,pshp(x),pshp(s),x=s;
}
void nsrt(int &x,LL w){
	if (!x) return void(t[x=++m]=(tree){w,w,rand(),1,1});
	t[x].sz++,t[x].s+=w;
	if (t[x].x==w) return void(t[x].p++);
	int f=t[x].x<w; nsrt(t[x].to[f],w);
	if (t[t[x].to[f]].r<t[x].r) rtt(x,f);
}
void dlt(int &x,LL w){
	if (!x) return; int l=t[x].to[0],r=t[x].to[1];
	if (t[x].x==w){
		if (t[x].p>1) return t[x].sz--,t[x].s-=w,void(t[x].p--);
		if (!l||!r) return void(x=l+r);
		rtt(x,(t[l].r<t[r].r)^1),dlt(x,w);
	}
	else t[x].sz--,t[x].s-=w,dlt(t[x].to[t[x].x<w],w);
}
int srch(int x,int w){
	if (!x) return 0; int l=t[x].to[0],r=t[x].to[1],p=t[x].p;
	if (t[l].sz>=w) return ans2+=t[x].x*p+t[r].s,srch(l,w);
	if (t[l].sz+t[x].p>=w){
		ans1+=t[l].s+t[x].x*(w-t[l].sz-1);
		ans2+=t[r].s+t[x].x*(t[l].sz+p-w);
		return x;	
	}
	return ans1+=t[x].x*p+t[l].s,srch(r,w-p-t[l].sz);
}
F void calc(int i){
	ans1=ans2=0; int d=k+1>>1,x=srch(rt,d);
	LL tmp=ans2-ans1+t[x].x*(d*2-1-k);
	if (tmp<ans) ans=tmp,mid=t[x].x,ls=i;
}
int main(){
	n=_read(),k=_read();
	for (int i=1;i<=n;i++) h[i]=_read();
	for (int i=1;i<=k;i++) nsrt(rt,h[i]);
	for (int i=k+1;i<=n;i++)
		calc(i-k),dlt(rt,h[i-k]),nsrt(rt,h[i]);
	calc(n+1-k),printf("%lld\n",ans);
	for (int i=1;i<ls;i++) printf("%d\n",h[i]);
	for (int i=ls;i<ls+k;i++) printf("%d\n",mid);
	for (int i=ls+k;i<=n;i++) printf("%d\n",h[i]);
	return 0;
}