这题好难啊……orz ZZK大佬,如果有一天我能和ZZK大佬一样强就好了……(不存在的!)
这题题目大意就是给出一个n长度的序列,从中选出k个长度在[l,r]范围内的互不相同的子序列,求这些子序列的和的最大值。
原来这题的模型叫超级钢琴啊,难道就是出自这题吗?
一般求最大值之类的东西,我们会很容易想到线段树啦,分治啦,其实还有一种求静态区间最值的高级数据结构——ST表。
先对原序列求一遍前缀和,把一个子序列的求和转化成两个端点权值的减法。
我们令MAX(i,l,r)表示右端点为i,左端点在[l,r]范围内使得sum[i]-sum[t-1]最大的位置t。
显然用堆来维护上面的三元组,每次挑出一个最大值,然后把那个三元组拆分成MAX(i,l,t-1)和MAX(i,t+1,r),重新插入堆中,这样做的目的是为了维护k个子序列互不相同。
最后k次取堆顶,累加统计答案即可。
附上AC代码:
#include <cstdio>
#include <cctype>
#include <cmath>
#include <queue>
using namespace std;
const int N=500010;
int n,m,l,r,a[N],sum[N],mn[N][21],x,y;
struct note{
int i,l,r,wz;
bool operator < (const note lyf) const {return sum[i]-sum[wz-1]<sum[lyf.i]-sum[lyf.wz-1];}
};
priority_queue <note> que;
long long ans;
inline char nc(void){
static char ch[100010],*p1=ch,*p2=ch;
return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;
}
inline void read(int &a){
static char c=nc();int f=1;
for (;!isdigit(c);c=nc()) if (c=='-') f=-1;
for (a=0;isdigit(c);a=a*10+c-'0',c=nc());
a*=f;return;
}
inline int get(int x,int y){return sum[x-1]<sum[y-1]?x:y;}
inline void build(void){
for (int j=1,k=log2(n); j<=k; ++j)
for (int i=1; i<=n; ++i)
if (i+(1<<j)-1<=n) mn[i][j]=get(mn[i][j-1],mn[i+(1<<j-1)][j-1]);
else mn[i][j]=mn[i][j-1];
}
inline int query(int x,int y){
int k=log2(y-x+1);
return get(mn[x][k],mn[y-(1<<k)+1][k]);
}
int main(void){
read(n),read(m),read(l),read(r);
for (int i=1; i<=n; ++i) read(a[i]),sum[i]=sum[i-1]+a[i],mn[i][0]=i;
build();
for (int i=l; i<=n; ++i) x=max(i-r+1,1),y=i-l+1,que.push((note){i,x,y,query(x,y)});
while (m--){
note p=que.top();que.pop();
ans+=(long long)sum[p.i]-sum[p.wz-1];
if (p.wz>p.l) que.push((note){p.i,p.l,p.wz-1,query(p.l,p.wz-1)});
if (p.wz<p.r) que.push((note){p.i,p.wz+1,p.r,query(p.wz+1,p.r)});
}
printf("%lld\n",ans);
return 0;
}