线段树 || BZOJ 1112: [POI2008]砖块Klo

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=1112

题解：

希望有连续K柱的高度是一样的，就先把1~K的数扔进线段树（线段树的下标就是数值，不需要离散化），求一波中位数和答案作为初始答案，

再从第K+1到N扫一遍，依次把每个数扔进线段树同时把第i-K个树弄出来扔掉，不断求中位数和更新答案就好了。

这里求序列中所有数到中位数的距离是这样求的：线段树多维护一个sum，当前序列中小于中位数的数的个数记为cnt1，

和为sum1，大于中位数的数的个数记为cnt2，和为sum2。于是答案就很显然是：pt（即中位数）*cnt1-sum1+sum2-pt*cnt2。

代码：

 #include<cstdio>
 #include<cstring>
 #include<iostream>
 #define ll long long
 #define min(a,b) ((a)<(b)?(a):(b))
 using namespace std;
 const int maxn=+,maxh=+,max_h=maxh-;
 int N,K,Z;
 ll sum[],H[maxn],ans,pt,cnt[];
 inline ll rd(){
     ll x=;int f=;char c=getchar();
     while(c<''||c>''){if(c=='-')f=-; c=getchar();}
     while(c>=''&&c<=''){x=x*+c-''; c=getchar();}
     return f*x;
 }
 struct Tree{
     int l,r;
     ll sum,cnt;
 }t[maxh<<];
 inline void Build(int x,int l,int r){
     t[x].l=l;t[x].r=r;int mid=(l+r)>>;
     if(l==r)return;
     Build(x<<,l,mid);Build(x<<|,mid+,r);
     return;
 }
 inline void Update(int x,int q,int o){
     int l=t[x].l,r=t[x].r,mid=(l+r)>>;
     if(l==r&&l==q){
         if(o==){t[x].sum+=l; t[x].cnt++;}
             else{t[x].sum-=l; t[x].cnt--;}
         return;
     }
     int ls=x<<,rs=x<<|;
     if(q<=mid)Update(ls,q,o);else Update(rs,q,o);
     t[x].sum=t[ls].sum+t[rs].sum;
     t[x].cnt=t[ls].cnt+t[rs].cnt;
     return;
 }
 inline int Find(int x,int z){//寻找中位数
     int l=t[x].l,r=t[x].r,ls=x<<,rs=x<<|;
     if(l==r)return l;
     if(t[ls].cnt>=z)return Find(ls,z);else return Find(rs,z-t[ls].cnt);
 }
 inline void Work(int x,int ql,int qr,int o){
     int l=t[x].l,r=t[x].r,mid=(l+r)>>,ls=x<<,rs=x<<|;
     if(ql<=l&&r<=qr){
         sum[o]+=t[x].sum;
         cnt[o]+=t[x].cnt;
         return;
     }
     if(ql<=mid)Work(ls,ql,qr,o);
     if(qr>mid)Work(rs,ql,qr,o);
     return;
 }
 int main(){
     N=rd();K=rd();
     for(int i=;i<=N;i++){H[i]=rd();H[i]++;}
     Build(,,max_h);
     for(int i=;i<=K;i++)Update(,H[i],);
     Z=(K+)>>;
     pt=Find(,Z);//pt即为中位数
     sum[]=sum[]=cnt[]=cnt[]=;
     Work(,,pt,);Work(,pt,max_h,);
     ans=(cnt[]*pt-sum[])+(sum[]-cnt[]*pt);
     for(int i=K+;i<=N;i++){
         Update(,H[i-K],);Update(,H[i],);
         pt=Find(,Z);
         sum[]=sum[]=cnt[]=cnt[]=;
         Work(,,pt,);Work(,pt,max_h,);
         if(ans>(cnt[]*pt-sum[])+(sum[]-cnt[]*pt)){
             ans=(cnt[]*pt-sum[])+(sum[]-cnt[]*pt);
         }
     }
     printf("%lld\n",ans);
     return ;
 }

---

By:AlenaNuna