【bzoj 1112】砖块Klo（权值线段树）

#include<bits/stdc++.h>
using namespace std;
const int inf=1000000;
int n,k,a[100005];
long long ans=1LL<<60;
struct Node{
Node *ch[2];
int siz;
long long sum;
Node();
inline void maintain(){
siz=ch[0]->siz+ch[1]->siz;
sum=ch[0]->sum+ch[1]->sum;
}
}*null=new Node,*root=null;
Node :: Node(){
ch[0]=ch[1]=null;
siz=sum=0;
}
void add(Node *&o,int l,int r,int x,int v){
if(o==null) o=new Node;
if(l==r){
o->siz+=v;  o->sum+=x;
return;
}
int mid=l+r>>1;
if(abs(x)<=mid) add(o->ch[0],l,mid,x,v);
else    add(o->ch[1],mid+1,r,x,v);
o->maintain();
}
int Kth(Node *o,int l,int r,int k){
if(l==r)    return l;
int mid=l+r>>1;
if(o->ch[0]->siz>=k)    return Kth(o->ch[0],l,mid,k);
else    return Kth(o->ch[1],mid+1,r,k-o->ch[0]->siz);
}
long long Query(Node *o,int l,int r,int p){
if(o==null)     return 0;
if(r<=p || l>p) return abs(o->sum-1ll*p*o->siz);
int mid=l+r>>1;
return Query(o->ch[0],l,mid,p)+Query(o->ch[1],mid+1,r,p);
}
int main(){
scanf("%d%d",&n,&k);
for(int i=1;i<=n;++i)   scanf("%d",&a[i]);
for(int i=1;i<=n;++i){
add(root,0,inf,a[i],1);
if(i-k>0)   add(root,0,inf,-a[i-k],-1);
if(i<k) continue;
int p=Kth(root,0,inf,(k+1)/2);
ans=min(ans,Query(root,0,inf,p));
}
printf("%lld\n",ans);
return 0;
}