【枚举】【权值分块】bzoj1112 [POI2008]砖块Klo

枚举长度为m的所有段，尝试用中位数更新答案。

所以需要数据结构，支持查询k大，以及大于/小于 k大值 的数的和。

平衡树、权值线段树、权值分块什么的随便呢。

 #include<cstdio>
 #include<algorithm>
 #include<cmath>
 using namespace std;
 typedef long long ll;
 struct Point{int v,p;}t[];
 bool operator < (const Point &a,const Point &b){return a.v<b.v;}
 int sumv[],ma[],en,a[],b[],tot[],l[],r[],num[],sum=,K,n,m;
 ll ans=,tAns;
 void makeblock()
 {
     int sz=sqrt(en); if(!sz) sz=;
     for(;sum*sz<en;++sum)
       {
           l[sum]=r[sum-]+; r[sum]=sum*sz;
           for(int i=l[sum];i<=r[sum];++i) num[i]=sum;
       }
     l[sum]=r[sum-]+; r[sum]=en;
     for(int i=l[sum];i<=r[sum];++i) num[i]=sum;
 }
 void Insert(const int &x){++b[x]; ++tot[num[x]]; sumv[num[x]]+=(ll)ma[x];}
 void Delete(const int &x){--b[x]; --tot[num[x]]; sumv[num[x]]-=(ll)ma[x];}
 int Query(const int &x)
 {
     int cnt=; ll now=;
     for(int i=;;i++)
       {
         cnt+=tot[i]; now+=sumv[i];
         if(cnt>=x)
           {
               cnt-=tot[i]; now-=sumv[i];
             for(int j=l[i];;j++)
               {
                 cnt+=b[j]; now+=(ll)ma[j]*(ll)b[j];
                 if(cnt>=x)
                   {
                       cnt-=b[j]; now-=(ll)ma[j]*(ll)b[j];
                       tAns=((ll)ma[j]*(ll)cnt-now);
                       return j;
                   }
               }
           }
       }
 }
 void Next_Sum(const int &x)
 {
     int cnt=; ll now=;
     for(int i=x+;i<=r[num[x]];++i) {cnt+=b[i]; now+=(ll)ma[i]*(ll)b[i];}
     for(int i=num[x]+;i<=sum;++i) {cnt+=tot[i]; now+=sumv[i];}
     tAns+=(now-(ll)ma[x]*(ll)cnt);
 }
 int main()
 {
     scanf("%d%d",&n,&m); K=(m>>)+;
     for(int i=;i<=n;++i)
       {
           scanf("%d",&t[i].v);
           t[i].p=i;
       } sort(t+,t+n+);
     ma[a[t[].p]=++en]=t[].v;
     for(int i=;i<=n;++i)
       {
           if(t[i].v!=t[i-].v) ++en;
           ma[a[t[i].p]=en]=t[i].v;
       } makeblock();
     for(int i=;i<=m;++i) Insert(a[i]);
     int t=Query(K); Next_Sum(t); ans=min(ans,tAns);
     for(int i=m+;i<=n;++i)
       {
           Delete(a[i-m]); Insert(a[i]);
           int t=Query(K);
         Next_Sum(t);
         ans=min(ans,tAns);
       } printf("%lld\n",ans);
     return ;
 }