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 100010
inline 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;
}