个人感觉这是一道ST表不错的题,可看出是让求k个在规定长度内的区间其和的总和最大值(不要说我语文差 = =)
可以搞一个结构体 存4个量 l,from,to,r,l表示区间左端点,[from,to]为区间右端点范围,r为区间右端点。
求一下前缀和(算是预处理吧)。题目中限制长度范围为[L,R]
#include<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
using namespace std;
const int M=500000+10;
int n,k,l,r;
int mn[M],st[M][20],val[M];
long long ans;
struct DATE{int le,fr,to,ri;};
priority_queue<DATE> pq;
inline bool operator < (DATE a,DATE b){return val[a.ri]-val[a.le-1]<val[b.ri]-val[b.le-1];}
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-'0';ch=getchar();}
return x*f;
}
inline void init(){ //ST表
mn[0]=-1;
for(int i=1;i<=n;i++){
mn[i]=(!(i&(i-1))) ? mn[i-1]+1 : mn[i-1];
st[i][0]=i;
}
for(int j=1;j<=mn[n];j++)
for(int i=1;i+(1<<j)-1<=n;i++){
int a=st[i][j-1],b=st[i+(1<<j-1)][j-1];
st[i][j]=(val[a]>=val[b] ? a : b);
}
}
inline int query(int l,int r){ //返回的是下标
int k=mn[r-l+1];
int a=st[l][k],b=st[r-(1<<k)+1][k];
return val[a]>val[b] ? a : b;
}
int main(){
n=read();k=read();l=read();r=read();
for(int i=1;i<=n;i++){val[i]=read();val[i]+=val[i-1];} //求出前缀
init();
for(int i=1;i+l-1<=n;i++){ //大块区间push进去
int rig=min(i+r-1,n);
pq.push((DATE){i,i+l-1,rig,query(i+l-1,rig)});
}
for(int i=1;i<=k;i++){
DATE zhuan=pq.top();pq.pop();
int mid=zhuan.ri;
ans+=val[zhuan.ri]-val[zhuan.le-1];
if(mid>zhuan.fr) pq.push((DATE){zhuan.le,zhuan.fr,mid-1,query(zhuan.fr,mid-1)});//再细找
if(mid<zhuan.to) pq.push((DATE){zhuan.le,mid+1,zhuan.to,query(mid+1,zhuan.to)});
}
printf("%lld",ans);
return 0;
}
。
先从1到n-L+1循环一遍,将所有满足要求的(i,i+L-1,min(i+R-1,n),query(i+L-1,min(i+R-1,n)))push进堆中。
然后在从1循环到k,取k个较大的数(当然不是直接取了)
每次取出堆顶,一定为当前最优解ans加上
然后将[from,to]分为[from,r-1] 与 [r+1,to] push进即可。
这样就可以保证取k次即为前k大的。
(ST表中存的是下表)
代码如下: