西北大学集训队选拔赛（重现赛）G 序列操作

链接：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
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;
}