【题解】[NOI2010] 超级钢琴 RMQ+优先队列 bzoj2006/洛谷 P2048

#include<bits/stdc++.h>
#define ll long long
#define inf 0x3f3f3f3f
#define Macth
#define max(x,y) ((x)>(y)?(x):(y))
#define min(x,y) ((x)<(y)?(x):(y))
const int N=500005,Log=20;
ll f[N][Log],sum[N];
template <typename Tp> inline void read(Tp &x){
    x=0;char ch=getchar();
    while(ch<'0'||ch>'9')ch=getchar();
    while(ch>='0'&&ch<='9')x=(x<<1)+(x<<3)+ch-'0',ch=getchar();
}
namespace RMQ{//ST 表求区间最优位置 (貌似在本题中是这样)
    inline void make(int n){
        for(register int i=1;i<=n;++i)f[i][0]=i;
        for(register int j=1;(1<<j)<=n;++j)
           for(register int i=1;i+(1<<j)-1<=n;++i){
             int x=f[i][j-1],y=f[i+(1<<(j-1))][j-1];
                f[i][j]=sum[x]>sum[y]?x:y;//取更优的位置
        }return;
    }
    inline int query(int l,int r){
        int k=log2(r-l+1);
        int x=f[l][k],y=f[r-(1<<k)+1][k];
        return sum[x]>sum[y]?x:y;
    }
}
int n,k,L,R;
struct Queue{
    int l,r,o,t;//t 就是最优的位置
    Queue(){}
    Queue(int o,int l,int r):o(o),l(l),r(r),t(RMQ::query(l,r)){}//t: 取个 l 至 r 区间的最优值
    bool operator < (Queue a)const{//重载运算符
        return sum[a.t]-sum[a.o-1]<sum[t]-sum[o-1];
    }
}A;
std::priority_queue<Queue> q;
int main(){
    scanf("%d%d%d%d",&n,&k,&L,&R);
    for(register int i=1;i<=n;++i){
        scanf("%lld",&sum[i]);sum[i]+=sum[i-1];
    }RMQ::make(n);ll ans=0;
    for(register int i=1;i<=n;++i){ 
       if(i+L-1<=n)q.push(Queue(i,i+L-1,min(i+R-1,n)));
    }while(k--){
        A=q.top();q.pop();
        ans+=sum[A.t]-sum[A.o-1];//更新 ans
        if(A.l!=A.t)q.push(Queue(A.o,A.l,A.t-1));
        if(A.r!=A.t)q.push(Queue(A.o,A.t+1,A.r));
    }printf("%lld\n",ans);
    return 0;
}