【BZOJ2006】超级钢琴（主席树，优先队列）
题面
题解
既然是一段区间
首先就要变成单点
所以求一个前缀和
这个时候贪心很明显了：
枚举每一个点和可以和它组成一段的可行的点
全部丢进一个堆里面
取出最大的\(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;
}