bzoj 2006: [NOI2010]超级钢琴 （st表+优先队列）

2006: [NOI2010]超级钢琴

Time Limit: 20 Sec   Memory Limit: 552 MB
Submit: 2563   Solved: 1267
[ Submit][ Status][ Discuss]

Description

小Z是一个小有名气的钢琴家，最近C博士送给了小Z一架超级钢琴，小Z希望能够用这架钢琴创作出世界上最美妙的

音乐。 这架超级钢琴可以弹奏出n个音符，编号为1至n。第i个音符的美妙度为Ai，其中Ai可正可负。 一个“超级

和弦”由若干个编号连续的音符组成，包含的音符个数不少于L且不多于R。我们定义超级和弦的美妙度为其包含的

所有音符的美妙度之和。两个超级和弦被认为是相同的，当且仅当这两个超级和弦所包含的音符集合是相同的。 

小Z决定创作一首由k个超级和弦组成的乐曲，为了使得乐曲更加动听，小Z要求该乐曲由k个不同的超级和弦组成。

我们定义一首乐曲的美妙度为其所包含的所有超级和弦的美妙度之和。小Z想知道他能够创作出来的乐曲美妙度最

大值是多少。

Input

第一行包含四个正整数n, k, L, R。其中n为音符的个数，k为乐曲所包含的超级和弦个数，L和R分别是超级和弦所

包含音符个数的下限和上限。 接下来n行，每行包含一个整数Ai，表示按编号从小到大每个音符的美妙度。

N<=500,000

k<=500,000

-1000<=Ai<=1000,1<=L<=R<=N且保证一定存在满足条件的乐曲

Output

只有一个整数，表示乐曲美妙度的最大值。

Sample Input

4 3 2 3
3
2
-6
8

Sample Output

11

【样例说明】
共有5种不同的超级和弦：
音符1 ~ 2，美妙度为3 + 2 = 5
音符2 ~ 3，美妙度为2 + (-6) = -4
音符3 ~ 4，美妙度为(-6) + 8 = 2
音符1 ~ 3，美妙度为3 + 2 + (-6) = -1
音符2 ~ 4，美妙度为2 + (-6) + 8 = 4
最优方案为：乐曲由和弦1，和弦3，和弦5组成，美妙度为5 + 2 + 4 = 11。

HINT

Source

[ Submit][ Status][ Discuss]

题解：st表+优先队列

我们先考虑如何求最大值，枚举每个起点，以该点位起点的合法方案一定是一段连续的区间。我预处理出前缀和，对于每个起点对答案的最大贡献其实就是以[i+l-1,i+r-1]为结尾的区间前缀和的最大值，但是我们现在要求前k大值。那么我们可以考虑经刚开始所以起点的最大值扔到大根堆中，向外弹出k次，每弹出一个点，就将以这个点开头，区间为[i+l-1,i+pos-1-1]和[i+pos+1-1,i+r-1]的最大值扔进队列，这样就可以解决了。

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<queue>
#define N 500003
#define LL long long
using namespace std;
int pos[N][30],len[N],n,k,l,r;
LL f[N][30],a[N],sum[N];
struct data  {
	int now,l,r,pos;
	LL val;
	bool operator <(const data &a) const {
		return val<a.val;
	} 
};
void solve()
{
	for (int i=1;i<=n;i++) f[i][0]=sum[i],pos[i][0]=i;
	for (int j=1;j<=19;j++)
	 for (int i=1;i<=n;i++)
	  if (i+(1<<j)-1<=n) {
	  	if (f[i][j-1]>f[i+(1<<(j-1))][j-1]) f[i][j]=f[i][j-1],pos[i][j]=pos[i][j-1];
	  	else f[i][j]=f[i+(1<<(j-1))][j-1],pos[i][j]=pos[i+(1<<(j-1))][j-1];
	  }
	int j=0;
	for (int i=1;i<=n;i++) {
		if ((1<<(j+1))<=i) j++;
		len[i]=j;
	}
}
void change(LL &val,int &pos1,int x,int y)
{
	int k1=len[y-x];
	if (f[x][k1]>f[y-(1<<k1)+1][k1]) pos1=pos[x][k1],val=f[x][k1];
	else pos1=pos[y-(1<<k1)+1][k1],val=f[y-(1<<k1)+1][k1]; 
}
int main()
{
	freopen("a.in","r",stdin);
	freopen("my.out","w",stdout);
	scanf("%d%d%d%d",&n,&k,&l,&r);
	for (int i=1;i<=n;i++) scanf("%I64d",&a[i]);
	for (int i=1;i<=n;i++) sum[i]=sum[i-1]+a[i];
	//for (int i=1;i<=n;i++) cout<<sum[i]<<" ";
    //cout<<endl;
	solve();
	priority_queue<data> p;
	for (int i=1;i<=n;i++){
		int x=i+l-1; int y=min(n,i+r-1);
		if (x>y) continue;
		data a; a.now=i; a.l=x; a.r=y; 
		change(a.val,a.pos,x,y);
		a.val=a.val-sum[a.now-1]; //cout<<a.val<<endl;
		p.push(a);
	}
	LL ans=0;
	for (int i=1;i<=k;i++) {
		data a=p.top(); p.pop();
		ans+=a.val;
		data b; 
		if (a.pos-1>=a.l) {
			b.now=a.now; b.l=a.l; b.r=a.pos-1;
			change(b.val,b.pos,b.l,b.r); b.val=b.val-sum[a.now-1]; //cout<<b.val<<endl;
			p.push(b);
		}
	    if (a.pos+1<=a.r) {
	    	b.now=a.now; b.l=a.pos+1; b.r=a.r;
	    	change(b.val,b.pos,b.l,b.r); b.val=b.val-sum[a.now-1]; //cout<<b.val<<endl;
			p.push(b);
		}
	}
	printf("%I64d\n",ans);
}