【BZOJ 1112】Poi2008砖块Klo treap

把每一个的高度看做是直线上的一个点，那么问题就是求一个点到其他所有点的距离最短。

可以发现，位于中位数的点到左右点的距离和最短，证明的话，很简单就不说了（试着把中位数上的点左右移动就好了）

那么问题就变成求中位数了，treap就搞定了。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<cstdlib>
#define LL long long
using namespace std;
struct Tree{
	Tree* ch[2];
	int r,size;LL val,sum;
	Tree(int x){
		r=rand();val=sum=x;size=1;
		ch[0]=ch[1]=NULL;
	}
	void up(){
		size=1;sum=val;
		if(ch[0]!=NULL)size+=ch[0]->size,sum+=ch[0]->sum;
		if(ch[1]!=NULL)size+=ch[1]->size,sum+=ch[1]->sum;
	}
}*rt;
void rotate(Tree*& u,int d){
	Tree* v=u->ch[!d];u->ch[!d]=v->ch[d];v->ch[d]=u;
	u->up();v->up();u=v;
}
void insert(Tree*& u,int x){
	if(u==NULL){u=new Tree(x);return;}
	int d;
	if(x>u->val)insert(u->ch[d=1],x);
	else insert(u->ch[d=0],x);
	if(u->ch[d]->r>u->r)rotate(u,!d);
	u->up();
}
void erase(Tree*& u,int x){
	if(u->val==x){
		if(u->ch[1]==NULL)u=u->ch[0];
		else if(u->ch[0]==NULL)u=u->ch[1];
		else{
			int d;
			if(u->ch[0]->r > u->ch[1]->r)rotate(u,d=1);
			else rotate(u,d=0);
			erase(u->ch[d],x);
		}
	}else if(u->val > x)erase(u->ch[0],x);
	else erase(u->ch[1],x);
	if(u!=NULL)u->up();
}
int n,K;LL h[100021],H,sum;
LL query(Tree* u,int k){
	int size=1;LL ans=u->val;
	if(u->ch[0]!=NULL)size+=u->ch[0]->size,ans+=u->ch[0]->sum;
	if(size==k){H=u->val;return ans;}
	if(size>k)return query(u->ch[0],k);
	else return ans+query(u->ch[1],k-size);
}
int main(){
	LL ans=1e15,ls;
	scanf("%d%d",&n,&K);
	for(int i=1;i<=K;i++)scanf("%d",h+i),insert(rt,h[i]),sum+=h[i];
	ls=query(rt,(K+1)/2);
	ans=min(ans,H*((K+1)/2)-ls+(sum-ls)-H*(K-(K+1)/2));
	for(int i=K+1;i<=n;i++){
		scanf("%d",h+i);
		sum=sum+h[i]-h[i-K];
		insert(rt,h[i]),erase(rt,h[i-K]);
		ls=query(rt,(K+1)/2);
		ans=min(ans,(K+1)/2*H-ls+(sum-ls)-(K-(K+1)/2)*H);
	}printf("%lld",ans);
	return 0;
}