题解 AT2264 【AGC008B】 Contiguous Repainting

一道结论题。

考虑最后一次操作，一定会将一段长为  的区间变成白色或黑色。

但是其余的操作，可以将其他的任何一个地方变成任意一种颜色。

所以只需要求 3 个值：区间和、区间之前所有正数的和、区间之后所有正数的和。

对于这个区间的值，如果为正数，就把他涂成黑色，如果为负数，就把他涂成白色。

CODE：

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define M 100005
ll n,k;
ll ans=-1;
ll a[M];
ll sum1[M],sum2[M],sum3[M];
int main(){
    scanf("%lld%lld",&n,&k);
    for(ll i=1;i<=n;i++){
        scanf("%lld",&a[i]);
        sum1[i]=sum1[i-1]+(a[i]<0?0:a[i]);//区间之前所有正数的和
        sum3[i]=sum3[i-1]+a[i];//前缀和求区间的和
    }
    for(ll i=n;i>=1;i--) sum2[i]=sum2[i+1]+(a[i]<0?0:a[i]);//区间之后所有正数的和
    for(ll i=1;i<=n-k+1;i++){
        ll A=sum1[i-1];
        ll B=sum2[i+k];
        ll C=sum3[i+k-1]-sum3[i-1];
        //求出这3个值
        if(C<0) C=0;
        ans=max(ans,A+B+C);
    }
    printf("%lld\n",ans);
    return 0;
}