数据结构优化dp是最骚的。。。
看到区间和,常用的技巧就是前缀和了。所以我们弄出一个前缀和数组b。
那么这个答案就能转化为\(max(a_i - a_j)\)。
我们每一次就只枚举i,那么这个max可以把\(a_i\)拿出来,也就是\(a_i - min(a_j)\)。
所以转换为区间最小值。直接转化为RMQ问题,使用ST表即可。
代码:
#include<cstdio>
#include<algorithm>
const int maxn = 500005;
int a[maxn], n, m;
int st[maxn][21];
int ans = -0x3f3f3f3f;
int query(int l, int r)
{
int k = 0;
while((1 << (k + 1)) <= r - l + 1) k++;
return std::min(st[l][k], st[r - (1 << k) + 1][k]);
}
int main()
{
scanf("%d%d", &n, &m);
for(int i = 1; i <= n; i++) scanf("%d", &a[i]);
for(int i = 1; i <= n; i++) a[i] += a[i - 1];
for(int i = 1; i <= n; i++) st[i][0] = a[i];
for(int j = 1; j <= 20; j++)
{
for(int i = 1; i + (1 << j) - 1 <= n; i++)
{
st[i][j] = std::min(st[i][j - 1], st[i + (1 << (j - 1))][j - 1]);
}
}
for(int i = 1; i <= n; i++)
{
int l = i - m;
if(l < 0) l = 0;
int temp = query(l, i);
ans = std::max(ans, a[i] - temp);
}
printf("%d\n", ans);
return 0;
}