考虑维护前缀和 $sum[i]$
那么对于每一个位置 $i$ ,左端点为 $i$ 右端点在 $[i+L-1,i+R-1]$ 区间的区间最大值容易维护
维护三元组 $(o,l,r)$ ,表示左端点为 $o$ ,右端点 $in [l,r]$ 的区间最大值,然后把它扔到一个堆里,每次弹出最大值计算贡献
计算完后,设此三元组右端点为 $t$,还要记得把 $(o,l,t-1)$,和 $(o,t+1,r)$ 扔到堆里
具体看代码,不难理解
#include<iostream> #include<cstdio> #include<algorithm> #include<cmath> #include<cstring> #include<queue> using namespace std; typedef long long ll; inline int read() { int x=0,f=1; char ch=getchar(); while(ch<'0'||ch>'9') { if(ch=='-') f=-1; ch=getchar(); } while(ch>='0'&&ch<='9') { x=(x<<1)+(x<<3)+(ch^48); ch=getchar(); } return x*f; } const int N=1e6+7; int n,K; int sum[N],f[N][21],Log[N]; namespace ST {//维护sum区间最大值的位置 void init() { for(int i=1;i<=n;i++) f[i][0]=i; Log[0]=-1; for(int i=1;i<=n;i++) Log[i]=Log[i>>1]+1; for(int k=1;(1<<k)<=n;k++) for(int i=1;i+(1<<k-1)<=n;i++) { if(sum[f[i][k-1]]>sum[ f[ i+(1<<k-1) ][k-1] ]) f[i][k]=f[i][k-1]; else f[i][k]=f[ i+(1<<k-1) ][k-1]; } } inline int query(int l,int r) { int k=Log[r-l+1]; if(sum[ f[l][k] ]>sum[ f[r-(1<<k)+1][k] ]) return f[l][k]; return f[r-(1<<k)+1][k]; } } struct dat { int o,l,r,t; dat (int o,int l,int r) : o(o),l(l),r(r),t(ST::query(l,r)) {} inline bool operator < (const dat &tmp) const { return sum[t]-sum[o-1]<sum[tmp.t]-sum[tmp.o-1]; } }; priority_queue <dat> Q; ll ans; int main() { n=read(),K=read(); int l=read(),r=read(),a; for(int i=1;i<=n;i++) a=read(),sum[i]=sum[i-1]+a; ST::init(); for(int i=1;i<=n;i++) if(i+l-1<=n) Q.push( dat(i,i+l-1,min(n,i+r-1)) ); while(K--) { dat T=Q.top(); Q.pop(); ans+=(sum[T.t]-sum[T.o-1]); if(T.l<T.t) Q.push(dat(T.o,T.l,T.t-1)); if(T.r>T.t) Q.push(dat(T.o,T.t+1,T.r)); } printf("%lld",ans); return 0; }