D. Yet Another Subarray Problem(cf 1197D)

题目传送门

由于m的值很小，可以枚举m种情况
原数列
1，2，3，4，5，6， m=3时
0，0，-k，0，0，-k
0，-k，0，0，-k，0
-k，0，0，-k，0，0 三种情况的数列对应加在一起 ，就枚举了所有的情况，做一下，前缀和处理

#include <iostream>
#include <stdio.h>
#include <string>
#include <string.h>
#include <map>
#include <math.h>
#include <vector>
#include <set>
#include <queue>
#include <map>
#include <deque>
using namespace std;
typedef long long ll;
const int N=3e5+50;
ll n,k,m;
ll a[N],b[N];
int main()
{
    scanf("%lld%lld%lld",&n,&m,&k);
    for(int i=1;i<=n;i++)
        scanf("%lld",&a[i]);
    ll ans=0;
    for(int i=0;i<m;i++)
    {
            for(int j=1;j<=n;j++)
                if(j%m==i)
                b[j]=a[j]-k;
                else
                    b[j]=a[j];
                for(int j=2;j<=n;j++)b[j]+=b[j-1];
                ll mi=0;
            for(int j=1;j<=n;j++){
 
                if(j%m==i)
                {
                    ans=max(ans,b[j]-mi);
                }
                mi=min(mi,b[j]);
 
            }
    }
    printf("%lld\n",ans);
    return 0;
}