【BZOJ 1112】Poi2008砖块Klo treap

来源：互联网发布：qq好友采集软件编辑：程序博客网时间：2024/04/25 20:03

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

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

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

#include<cstdio>#include<cstring>#include<iostream>#include<cstdlib>#define LL long longusing 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;}

0 0