链接:https://ac.nowcoder.com/acm/contest/892/G
来源:牛客网
题目描述
一个长度为N的序列,所有元素均大于等于 −M−M,你可以修改任意元素的值,使得修改之后所有元素的值仍均大于等于−M−M,且所有长度为k的区间的元素和均为负数,求最小的改动量之和。设一个元素aiai改动后为bibi,则这个元素的改动量为|ai−bi||ai−bi|
输入描述:
第一行包含三个正整数N,M,K,意义如题面所示。1≤K≤N≤2∗1061≤K≤N≤2∗106,1≤M≤1071≤M≤107
第二行包含N个整数a,描述这个序列的每一个元素大小−M≤a≤107−M≤a≤107
输出描述:
输出一行一个整数表示答案。
示例1
输入
复制
4 5 2 7 5 8 10
输出
复制
32
说明
可改为0 -1 0 -1,满足题目要求,且改动量之和为32
考虑贪心,对于每一个长度为k的区间,从后面往前修改,尽量改成-m,除非区间和为正了,改成使得区间和为-1的数
我这里用的树状数组维护区间查询及单点修改
#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int maxn=2e6+10;
ll c[maxn];int a[maxn];int n,m,k;ll ans;
inline int lowbit(int x){return x&(-x);}
inline void update(int i,int val){while(i<=n){c[i]+=val;i+=lowbit(i);}}
inline ll query(int i){ll res=0;while(i>0){res+=c[i];i-=lowbit(i);}return res;}
inline ll Query(int l,int r){return query(r)-query(l-1);}
inline void read(int &X)
{
X=0;int w=0;char ch=0;
while(!isdigit(ch))w|=ch=='-',ch=getchar();
while( isdigit(ch))X=(X<<3)+(X<<1)+(ch^48),ch=getchar();
X=w?-X:X;
}
int main()
{
scanf("%d%d%d",&n,&m,&k);ll ans=0;
for(int i=1;i<=n;++i)read(a[i]),update(i,a[i]);
for(int i=k;i<=n;++i)
{
ll tmp=Query(i-k+1,i);
if(tmp>=0)
{
for(int j=i;j>=i-k+1;--j)
{
if(tmp-a[j]-m>=0)
{
tmp=tmp-a[j]-m;
update(j,-m-a[j]);
ans+=abs(a[j]+m);
}
else
{
ans+=abs(tmp+1);
update(j,-tmp-1);
break;
}
}
}
}
printf("%lld\n",ans);
return 0;
}