【BZOJ2006】超级钢琴（主席树，优先队列）

【BZOJ2006】超级钢琴（主席树，优先队列）

题面

BZOJ

题解

既然是一段区间

首先就要变成单点

所以求一个前缀和

这个时候贪心很明显了：

枚举每一个点和可以和它组成一段的可行的点

全部丢进一个堆里面

取出最大的\(K\)个就行了

但是，很显然，我们做不到都取出来

所以，考虑怎么优化这个过程

每次堆里面对于每个点就先维护一个最大贡献

显然的，只有取出了最大贡献，才会取出次大贡献

那么，最大/次大/\(K\)大贡献怎么算？

一个点，能够和他组成和弦的是一个连续的区间

最大的贡献就是它和区间最小值的差

依次类推

就是区间第\(K\)大

所以搞一棵主席树

维护一下区间第\(K\)大就好啦

再用堆每次取出/加入即可

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<set>
#include<map>
#include<vector>
#include<queue>
using namespace std;
#define MAX 510000
inline int read()
{
	int x=0,t=1;char ch=getchar();
	while((ch<'0'||ch>'9')&&ch!='-')ch=getchar();
	if(ch=='-')t=-1,ch=getchar();
	while(ch<='9'&&ch>='0')x=x*10+ch-48,ch=getchar();
	return x*t;
}
struct Data{int v,id,k;};
bool operator<(Data a,Data b){return a.v<b.v;}
priority_queue<Data> Q;
struct Node
{
	int ls,rs;
	int sum;
}t[MAX<<6];
int tot,S[MAX],top,rt[MAX],L,R,n,K;
int a[MAX];
void Modify(int &now,int ff,int l,int r,int pos,int w)
{
	t[now=++tot]=t[ff];t[now].sum+=w;
	if(l==r)return;
	int mid=(l+r)>>1;
	if(pos<=mid)Modify(t[now].ls,t[ff].ls,l,mid,pos,w);
	else Modify(t[now].rs,t[ff].rs,mid+1,r,pos,w);
}
int Query(int A,int B,int l,int r,int K)
{
	if(l==r)return l;
	int mid=(l+r)>>1;
	int sum=t[t[A].ls].sum-t[t[B].ls].sum;
	if(sum<K)return Query(t[A].rs,t[B].rs,mid+1,r,K-sum);
	else return Query(t[A].ls,t[B].ls,l,mid,K);
}
int main()
{
	n=read()+1;K=read();L=read();R=read();
	for(int i=2;i<=n;++i)S[++top]=a[i]=a[i-1]+read();S[++top]=0;
	sort(&S[1],&S[top+1]);
	top=unique(&S[1],&S[top+1])-S-1;
	for(int i=1;i<=n;++i)a[i]=lower_bound(&S[1],&S[top+1],a[i])-S;
	for(int i=1;i<=n;++i)Modify(rt[i],rt[i-1],1,top,a[i],1);
	for(int i=L+1;i<=n;++i)
	{
		Data x;x.id=i;x.k=1;
		x.v=S[a[i]]-S[Query(rt[i-L],rt[max(0,i-R-1)],1,top,1)];
		Q.push(x);
	}
	long long ans=0;
	while(K--)
	{
		Data x=Q.top();Q.pop();
		ans+=x.v;
		if(x.k==min(x.id-L,R-L+1))continue;
		x.k++;
		x.v=S[a[x.id]]-S[Query(rt[x.id-L],rt[max(0,x.id-R-1)],1,top,x.k)];
		Q.push(x);
	}
	printf("%lld\n",ans);
	return 0;
}