【bzoj2006】NOI2010超级钢琴

补了下前置技能……

题意就是求一段区间的权值和前k大的子序列的和。

把段扔进优先队列

每次拿出来之后按照所选择的j进行分裂

#include<bits/stdc++.h>
#define N 500005
#define mp(a,b,c,d) (seg){a,b,c,d}
using namespace std;
typedef long long ll;
struct seg{int i,l,r,t;};
int bin[],Log[N],n,k,l,r,a[N],st[][N];
ll ans=;
void initrmq(){
    Log[]=-;for(int i=;i<=n;i++)Log[i]=Log[i>>]+;
    for(int i=;i<=n;i++)st[][i]=i;
    for(int i=n;i;i--)for(int j=;j<=;j++)
    if(i+bin[j]-<=n){
        int t1=st[j-][i],t2=st[j-][i+bin[j-]];
        st[j][i]=a[t1]>a[t2]?t1:t2;
    }
    else break;
}
inline int query(int l,int r){
    if(l==r)return l;int t=Log[r-l+];
    int t1=st[t][l],t2=st[t][r-bin[t]+];
    return a[t1]>a[t2]?t1:t2;
}
inline bool operator <(seg x,seg y){
    return a[x.t]-a[x.i-]<a[y.t]-a[y.i-];
}
inline int read(){
    int f=,x=;char ch;
    do{ch=getchar();if(ch=='-')f=-;}while(ch<''||ch>'');
    do{x=x*+ch-'';ch=getchar();}while(ch>=''&&ch<='');
    return f*x;
}
int main(){
    bin[]=;for(int i=;i<=;i++)bin[i]=bin[i-]<<;
    n=read();k=read();l=read();r=read();
    for(int i=;i<=n;i++)a[i]=read()+a[i-];
    initrmq();
    priority_queue<seg,vector<seg> >q;
    for(int i=;i<=n;i++)if(i+l-<=n){
        int t=min(n,i+r-);q.push(mp(i,i+l-,t,query(i+l-,t)));
    }
    for(int i=;i<=k;i++){
        seg t=q.top();q.pop();
        ans+=a[t.t]-a[t.i-];
        if(t.t->=t.l)q.push(mp(t.i,t.l,t.t-,query(t.l,t.t-)));
        if(t.t+<=t.r)q.push(mp(t.i,t.t+,t.r,query(t.t+,t.r)));
    }
    cout<<ans<<endl;
}