洛谷 P2048 [NOI2010]超级钢琴（优先队列，RMQ）

传送门

我们定义$(p,l,r)=max\{sum[t]-sum[p-1],p+l-1\leq t\leq p+r-1 \}$

那么因为对每一个$p$来说$sum[p-1]$是一个定值，所以我们只要在$[p+l-1,p+r-1]$的区间里找出最大的$sum[t]$就行了，这就是一个RMQ问题，开个ST表就行了

一开始我们用优先队列存储所有的$(p,l,r)$，然后每一次取出队首更新答案

注意如果$t$被选了，答案还有可能在$t$两边的区间里，所以记得把$(p,l,t-1)$和$(p,t+1,r)$给扔进优先队列里

 //minamoto
 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #include<queue>
 #include<cmath>
 #define ll long long
 using namespace std;
 #define getc() (p1==p2&&(p2=(p1=buf)+fread(buf,1,1<<21,stdin),p1==p2)?EOF:*p1++)
 char buf[<<],*p1=buf,*p2=buf;
 inline int read(){
     #define num ch-'0'
     char ch;bool flag=;int res;
     while(!isdigit(ch=getc()))
     (ch=='-')&&(flag=true);
     for(res=num;isdigit(ch=getc());res=res*+num);
     (flag)&&(res=-res);
     #undef num
     return res;
 }
 const int N=,LOG=;
 ll sum[N],st[N][LOG];
 int n,k,l,r;
 void init(){
     for(int i=;i<=n;++i) st[i][]=i;
     for(int j=;(<<j)<=n;++j)
     for(int i=;i+(<<j)-<=n;++i){
         int x=st[i][j-],y=st[i+(<<j-)][j-];
         st[i][j]=sum[x]>sum[y]?x:y;
     }
 }
 inline int query(int l,int r){
     int k=log2(r-l+);
     int x=st[l][k],y=st[r-(<<k)+][k];
     return sum[x]>sum[y]?x:y;
 }
 struct node{
     int p,l,r,t;
     node(){}
     node(int p,int l,int r):p(p),l(l),r(r),t(query(l,r)){}
     inline bool operator <(const node &b)const
     {return sum[t]-sum[p-]<sum[b.t]-sum[b.p-];}
 };
 priority_queue<node> q;
 int main(){
 //    freopen("testdata.in","r",stdin);
     n=read(),k=read(),l=read(),r=read();
     for(int i=;i<=n;++i)
     sum[i]=read(),sum[i]+=sum[i-];
     init();
     for(int i=;i<=n;++i)
     if(i+l-<=n)
     q.push(node(i,i+l-,min(i+r-,n)));
     ll ans=;
     while(k--){
         int p=q.top().p,l=q.top().l,r=q.top().r,t=q.top().t;
         q.pop();
         ans+=sum[t]-sum[p-];
         if(l!=t) q.push(node(p,l,t-));
         if(t!=r) q.push(node(p,t+,r));
     }
     printf("%lld\n",ans);
     return ;
 }