【BZOJ】2006: [NOI2010]超级钢琴（前缀和+RMQ+堆）

题目

传送门：QWQ

分析

又不会做。。。。。。。

显然很好想到前缀和处理一下。

然后考虑最大化结果，直接上st表。

问题来了，然后呢？

怎么做$ length \in [l,r]  $ 呢？

正解是设一个五元组 (i,l,r,val,pos) 。

i是左端点，l,r是右端点范围，val是 i 到 pos的和, pos是 右端点位置。

 然后对于  l,r   二分，扔进优先队列处理 。取出前 k 大，累加一下就是答案。

完了。

代码

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn=;
int n,posi[maxn][];
int sum[maxn],Log2[maxn];
struct Node{
    int i,l,r,val,pos;
    bool operator < (const Node& a) const{ return val<a.val;}
};

inline int Max(int x,int y) {return sum[x]>sum[y]? x:y;}
void ST(){
    Log2[]=-;
    for(int i=;i<=n;i++)
        if((i&(i-))==) Log2[i]=Log2[i-]+; else Log2[i]=Log2[i-];
    for(int i=;i<=n;i++) posi[i][]=i;

    for(int j=;(<<j)<=n;j++)
        for(int i=;i+(<<j)<=n+;i++){
            posi[i][j]=Max(posi[i][j-],posi[i+(<<(j-))][j-]);
        }
}
inline int RMQ(int l,int r){
    int tmp=Log2[r-l+];
    return Max(posi[l][tmp],posi[r-(<<tmp)+][tmp]);
}
inline int read()
{
    int x=,f=; char ch=getchar();
    while (ch<'' || ch>'') {if (ch=='-') f=-; ch=getchar();}
    while (ch>='' && ch<='') {x=x*+ch-''; ch=getchar();}
    return x*f;
}
priority_queue<Node> que;
int main()
{
    int k,L,R;
    scanf("%d%d%d%d",&n,&k,&L,&R);
    for(int i=;i<=n;i++){
        sum[i]=sum[i-]+read();
    }
    ST();
    for(int i=;i<=n+-L;i++){
        int pos=RMQ(i+L-,min(n,i+R-));
//        printf("---------- %d %d %d\n",i+L-1,min(n,i+R-1),pos);
        que.push((Node){i,i+L-,min(n,i+R-),sum[pos]-sum[i-],pos});
    }
    ll ans=;
    while(k){
        Node x=que.top();que.pop();
//        printf("##########  %d %d %d\n",x.i,x.pos,x.val);
        k--; ans+=(ll)x.val;
        Node ls=x,rs=x;
        ls.r=x.pos-; rs.l=x.pos+;
        if(ls.r>=ls.l) {
            ls.pos=RMQ(ls.l,ls.r); ls.val=sum[ls.pos]-sum[ls.i-]; que.push(ls);
        }
        if(rs.r>=rs.l) {
            rs.pos=RMQ(rs.l,rs.r); rs.val=sum[rs.pos]-sum[rs.i-]; que.push(rs);
        }
    }
//    puts("-----------debug------------");
//    for(int i=1;i<=n;i++)    printf("%d ",Log2[i]);
    printf("%lld\n",ans);
    return ;
}