bzoj 1112 [POI2008]砖块Klo

Description

N柱砖,希望有连续K柱的高度是一样的. 你可以选择以下两个动作 1:从某柱砖的顶端拿一块砖出来,丢掉不要了. 2:从仓库中拿出一块砖,放到另一柱.仓库无限大. 现在希望用最小次数的动作完成任务.

Input

第一行给出N,K. (1 ≤ k ≤ n ≤ 100000), 下面N行,每行代表这柱砖的高度.0 ≤ hi ≤ 1000000

Output

最小的动作次数

Sample Input

5 3

3

9

2

3

1

好题啊QAQ

所以我们一句话题意 在一个固定长度的区间内 使之都变为中位数的代价的绝对值之和。

嗯没错就是这样。。

所以为什么是中位数？ 唔 小学奥数？

然后枚举区间 k 然后 找区间中位数 用两个权值树状数组 维护就ok了

注意树状数组是不能存0的 所以要+1 存储

唔 代码不是 很长 实现的还不错

#include<cstdio>
#include<cstring>
#include<algorithm>
#define int long long
#define maxm  1000006
using namespace std;
//by mars_ch
int n,k;
int h[maxm];
struct data
{
    int c[maxm];
    int lowbit(int x)
    {
        return x&(-x);
    }
    void modify(int x,int y)
    {
        while(x<=1000005)
        {
            c[x]+=y;
            x+=lowbit(x);
        }
    }
    int query(int x)
    {
        int res=0;
        while(x)
        {
            res+=c[x];
            x-=lowbit(x);
        }
        return res;
    }
}cnt,sum;
int binary_search(int x)
{
    int l=1,r=maxm-1,res=1;
    while(l<=r)
    {
        int mid=(l+r)/2;
        if(cnt.query(mid)>=x)
        {
            r=mid-1;
            res=mid;
        }
        else l=mid+1;
    }
    return res;
}
signed main()
{
    scanf("%lld%lld",&n,&k);
    for(int i=1;i<=n;i++)
    {
        scanf("%lld",&h[i]);
        h[i]++;     //树状数组不能存0 
    }
    for(int i=1;i<k;i++)
    {
        cnt.modify(h[i],1);
        sum.modify(h[i],h[i]);
    }
    int kk=(k+1)/2;
    int ans=0x3fffffffffff;
    for(int i=k;i<=n;i++)
    {
        cnt.modify(h[i],1);
        sum.modify(h[i],h[i]);
        int id=binary_search(kk);

        ans=min(ans,id*cnt.query(id)-sum.query(id)+sum.query(maxm-1)-sum.query(id)-id*(k-cnt.query(id)));

        cnt.modify(h[i-k+1],-1);
        sum.modify(h[i-k+1],-h[i-k+1]);
     } 

     printf("%lld\n",ans);
    return 0;
}