bzoj1112 [POI2008]砖块Klo(平衡树/map/树状数组)

把一个区间的数都变成一个数，显然都变成中位数需要的次数最少，而次数就是后k/2的数的和-前k/2的数的和，因此原问题就是求区间中位数。区间中位数可以平衡树来搞，也可以用stl水过hh。先把1…k排序，前k/2个数扔进mp1,剩下的数扔进mp2,此时mp2中的最小值就是中位数mid。然后我们维护这两个map，以及这两个map的大小，和，即可。每次先删掉x,再扔进新的y，根据两个map的size把两个map调整对（把mp1的最大值扔进mp2，mp2的最小值扔进mp1什么的）即可。

#include <bits/stdc++.h>using namespace std;#define ll long long#define inf 0x3f3f3f3f#define N 100010inline int read(){    int x=0,f=1;char ch=getchar();    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();    return x*f;}int n,k,a[N],b[N],sz1=0,sz2=0;map<int,int>mp1,mp2;ll ans=0,sum1=0,sum2=0;int main(){//  freopen("a.in","r",stdin);    n=read();k=read();for(int i=1;i<=n;++i) a[i]=read();    for(int i=1;i<=k;++i) b[i]=a[i];sort(b+1,b+k+1);    for(int i=1;i<=k/2;++i) mp1[b[i]]++,sz1++,sum1+=b[i];    for(int i=k/2+1;i<=k;++i) mp2[b[i]]++,sz2++,sum2+=b[i];    int mid=mp2.begin()->first;ans=sum2-sum1-k%2*mid;    for(int i=2;i+k-1<=n;++i){        int x=a[i-1],y=a[i+k-1];        if(x>=mid){if(--mp2[x]==0) mp2.erase(x);sz2--;sum2-=x;}        else{if(--mp1[x]==0) mp1.erase(x);sz1--;sum1-=x;}        if(y>=mid){mp2[y]++;sz2++;sum2+=y;}        else{mp1[y]++;sz1++;sum1+=y;}        if(sz2-sz1>k%2){            int x=mp2.begin()->first;            if(--mp2[x]==0) mp2.erase(x);sz2--;sum2-=x;            mp1[x]++;sz1++;sum1+=x;        }else if(sz2-sz1<k%2){            int x=(--mp1.end())->first;            if(--mp1[x]==0) mp1.erase(x);sz1--;sum1-=x;            mp2[x]++;sz2++;sum2+=x;        }mid=mp2.begin()->first;ans=min(ans,sum2-sum1-k%2*mid);    }printf("%lld\n",ans);    return 0;}