【BZOJ】2006: [NOI2010]超级钢琴-优先队列

传送门:bzoj2006

---

题解

实际上是道水题，但还是debug了很久。
先求前缀和。用ST表存一下区间里前缀和最大的。枚举左端点，逐个加入优先队列。
我们总共会取k次，每次取出后，由于考虑到以该点为左端点的其他右端点所形成的次大区间还可能对答案有贡献，所以把区间拆成左边一段（取左边一段的最优值），右边一段（取右边一段的最优值）（以此次取出的这个区间的右端点为分界线，右端点不加入左边也不加入右边的区间）。

---

代码

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cstring>
#include<cctype>
#include<queue>
#define inf 0x7fffffff
using namespace std;
typedef long long ll;
const int N=5e5+10;
ll ans;
int n,m,k,L,R,st[N][22],bin[22],Log[N],sum[N],Rag,id;
inline int imax(ll x,ll y){return x>y?x:y;}
inline int imin(int x,int y){return x>y?y:x;}
struct P{
    int v,pos,l,r;
}tmp,ret;   
bool operator <(const  P& a,const P& b){return a.v<b.v;}
priority_queue<P>Q;
char ch;
inline int rd(){
    int x=0,f=1;
    while(!isdigit(ch=getchar())) if(ch=='-')f=-1;
    x=(ch^48);
    while(isdigit(ch=getchar())){x=x*10+(ch^48);}
    return f*x;
}
inline ll query(int l,int r)
{
     Rag=Log[r-l+1];
     l=st[l][Rag],r=st[r-bin[Rag]+1][Rag];
     return sum[l]>sum[r]? l:r;
}

int main(){
    bin[0]=1;for(int i=1;i<22;++i) bin[i]=bin[i-1]<<1;
    n=rd();k=rd();L=rd();R=rd();    
    for(int i=2;i<=n;++i) 
     Log[i]=Log[i>>1]+1;
    for(int now,i=1;i<=n;++i){sum[i]=sum[i-1]+rd();st[i][0]=i;}
    for(int x,y,j,i=1;bin[i]<=n;++i){
        for(j=1;j+bin[i]<=n+1;++j){
            x=st[j][i-1],y=st[j+bin[i-1]][i-1];
            st[j][i]= sum[x]>sum[y]? x:y;
        }
    }
    for(int ql,qr,i=1;i+L<=n+1;++i) {
        ret.v=sum[query(i+L-1,imin(n,i+R-1))]-sum[i-1];
        ret.pos=i;ret.l=i+L-1;ret.r=imin(i+R-1,n);
        Q.push(ret);
    }
    for(int i=1;i<=k;i++){
        tmp=Q.top();Q.pop();
        id=query(tmp.l,tmp.r);ans+=tmp.v;
        if(id>tmp.l) {
            ret.v=sum[query(tmp.l,id-1)]-sum[tmp.pos-1];ret.pos=tmp.pos;
            ret.l=tmp.l;ret.r=id-1;Q.push(ret);
        }
        if(id<tmp.r) {
            ret.v=sum[query(id+1,tmp.r)]-sum[tmp.pos-1];
            ret.pos=tmp.pos;ret.l=id+1;ret.r=tmp.r;Q.push(ret);
        }
    }
    printf("%lld\n",ans);
}