题目
其实求的就是最大连续子段和,长度在p到q之间。
分析
然后就用RMQ,改成前缀和。
状态转移方程:
f
[
i
]
[
j
]
=
m
a
x
(
f
[
i
]
[
j
−
1
]
,
f
[
i
+
2
j
−
1
]
[
j
−
1
]
)
f[i][j]=max(f[i][j-1],f[i+2^{j-1}][j-1])
f[i][j]=max(f[i][j−1],f[i+2j−1][j−1])
代码
#include <cstdio>
#include <cmath>
#include <algorithm>
using namespace std;
long long f[100001][21],ma=-1e18; int n,p,q;
int main(){
scanf("%d%d%d",&n,&p,&q);
for (int i=1;i<=n;i++) scanf("%lld",&f[i][0]),f[i][0]+=f[i-1][0];
for (int j=1;(1<<j)<=n;j++)
for (int i=1;i+(1<<j)-1<=n;i++)
f[i][j]=max(f[i][j-1],f[i+(1<<j-1)][j-1]);
for (int i=1;i<=n-p+1;i++){
int l=i+p-1,r=min(n,q+i-1),z;
z=(int)(log(r-l+1)/log(2));
ma=max(ma,max(f[l][z],f[r+1-(1<<z)][z])-f[i-1][0]);
}
printf("%lld",ma);
return 0;
}