【BZOJ】2006 [NOI2010]超级钢琴 ST表+堆+贪心

题目传送门

这题好难啊……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;
}