那么问题就在于如何找任意区间的中位数
划分树和平衡树都可以解决
所以是水题一道……
下面描述我的悲惨经历……
上周周末就看到了这道题
觉得应该可以用划分树水过
屁颠屁颠写了个划分树,写到一半发现不对头,好像不能做
和pty讨论了一下发现实际上还是可以做
只不过维护很蛋疼……
然后因为要期末考试了,把它放在那里没管
今天有空打算写一下
然后想的时候又觉得划分树没法做,然后就被绕进去了……
果断把代码一把删掉……
结果实际上划分树是真的可以做……
然后想一想还是写平衡树好了
然后再想一想发现multiset可以水过
虽然STL不支持找第k大 //也许是出于对速度的保证,STL并没有维护sum,size等“多余”的信息
但是因为新加进来一个数中位数只会向前向后挪一位,因此还是可求的
写的时候发现各种菊紧,比如说删掉了中位数之类的,STL要各种特判
然后火了一把删掉,手写Splay
写Splay的时候心血来潮想把它写成class的形式
许久没写Splay结果zigzag差点写疵
以前一般都写的静态Splay,这里来个动态的让我有点小纠结……
然后写啊写啊写……
写完了,wa啊wa啊wa……
最后发现Del过程写疵了……
首先是删点的时候size没有正确维护……
还有就是如果只有一个点的话,不能正确删除……//我没有判断把要删除的点旋到根以后左右子树都为空的情况……
结果k=1的点统统tle……
以上……
//Lib #include<cstdio> #include<cstring> #include<cstdlib> #include<cmath> #include<ctime> #include<iostream> #include<algorithm> #include<vector> #include<string> #include<queue> #include<set> #include<map> using namespace std; //Macro #define rep(i,a,b) for(int i=a,tt=b;i<=tt;++i) #define drep(i,a,b) for(int i=a,tt=b;i>=tt;--i) #define erep(i,e,x) for(int i=x;i;i=e[i].next) #define irep(i,x) for(__typedef(x.begin()) i=x.begin();i!=x.end();i++) #define read() (strtol(ipos,&ipos,10)) #define sqr(x) ((x)*(x)) #define pb push_back #define PS system("pause"); typedef long long ll; typedef pair<int,int> pii; const int oo=~0U>>1; const double inf=1e100; const double eps=1e-6; string name="", in=".in", out=".out"; //Var struct T { int LC,RC,FA,SIZE; ll SUM,KEY; #define lc(x) tree[x].LC #define rc(x) tree[x].RC #define fa(x) tree[x].FA #define size(x) tree[x].SIZE #define sum(x) tree[x].SUM #define key(x) tree[x].KEY }tree[100008]; int tot,root; int n,k; ll ans,num[100008]; void Update(int x) { sum(x)=sum(lc(x))+sum(rc(x))+key(x); size(x)=size(lc(x))+size(rc(x))+1; } void Zig(int x) { int y=fa(x); if(lc(fa(y))==y)lc(fa(y))=x;else rc(fa(y))=x;fa(x)=fa(y); fa(rc(x))=y;lc(y)=rc(x);rc(x)=y;fa(y)=x; Update(y); } void Zag(int x) { int y=fa(x); if(lc(fa(y))==y)lc(fa(y))=x;else rc(fa(y))=x;fa(x)=fa(y); fa(lc(x))=y;rc(y)=lc(x);lc(x)=y;fa(y)=x; Update(y); } void Splay(int x) { for(int y,z;fa(x);) { y=fa(x);z=fa(y); if(!z) if(lc(y)==x)Zig(x); else Zag(x); else if(lc(z)==y) if(lc(y)==x)Zig(y),Zig(x); else Zag(x),Zig(x); else if(rc(y)==x)Zag(y),Zag(x); else Zig(x),Zag(x); } root=x; Update(x); } int Find(int k) { int p=root; while(1) { if(size(lc(p))+1==k)return p; else if(size(lc(p))+1>k)p=lc(p); else k-=size(lc(p))+1,p=rc(p); } } void Ins(ll x) { int f=0,p=root,flag=0; while(1) { if(p==0) { key(++tot)=x,fa(tot)=f; if(flag)rc(f)=tot;else lc(f)=tot; Splay(tot); break; } if(key(p)>=x)f=p,p=lc(p),flag=0; else f=p,p=rc(p),flag=1; } } void Del(int p) { int q; Splay(p); if(!lc(p)&&!rc(p)){root=0;return;} if(lc(p)==0){fa(rc(p))=0;Splay(rc(p));} else if(rc(p)==0){fa(lc(p))=0;Splay(lc(p));} else { fa(lc(p))=0; for(q=lc(p);rc(q);q=rc(q)); Splay(q); fa(rc(p))=q;rc(q)=rc(p); Update(q); } } ll Calc() { int size=size(root)+1>>1; int p=Find(size); Splay(p); return key(p)*size(lc(p))-sum(lc(p))+sum(rc(p))-key(p)*size(rc(p)); } void Init() { ios::sync_with_stdio(0); cin>>n>>k; rep(i,1,n)cin>>num[i]; } void Work() { rep(i,1,k) Ins(num[i]); ans=Calc(); rep(i,k+1,n) { Del(i-k); Ins(num[i]); ans=min(ans,Calc()); } cout<<ans<<endl; } int main() { // freopen((name+in).c_str(),"r",stdin); // freopen((name+out).c_str(),"w",stdout); Init(); Work(); return 0; }