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