BZOJ 2006: [NOI2010]超级钢琴

Description

小Z是一个小有名气的钢琴家，最近C博士送给了小Z一架超级钢琴，小Z希望能够用这架钢琴创作出世界上最美妙的

音乐。 这架超级钢琴可以弹奏出n个音符，编号为1至n。第i个音符的美妙度为Ai，其中Ai可正可负。 一个“超级

和弦”由若干个编号连续的音符组成，包含的音符个数不少于L且不多于R。我们定义超级和弦的美妙度为其包含的

所有音符的美妙度之和。两个超级和弦被认为是相同的，当且仅当这两个超级和弦所包含的音符集合是相同的。 

小Z决定创作一首由k个超级和弦组成的乐曲，为了使得乐曲更加动听，小Z要求该乐曲由k个不同的超级和弦组成。

我们定义一首乐曲的美妙度为其所包含的所有超级和弦的美妙度之和。小Z想知道他能够创作出来的乐曲美妙度最

大值是多少。

Input

第一行包含四个正整数n, k, L, R。其中n为音符的个数，k为乐曲所包含的超级和弦个数，L和R分别是超级和弦所

包含音符个数的下限和上限。 接下来n行，每行包含一个整数Ai，表示按编号从小到大每个音符的美妙度。

N<=500,000

k<=500,000

-1000<=Ai<=1000,1<=L<=R<=N且保证一定存在满足条件的乐曲

Output

只有一个整数，表示乐曲美妙度的最大值。

Sample Input

4 3 2 3
3
2
-6
8

Sample Output

11

【样例说明】
共有5种不同的超级和弦：
音符1 ~ 2，美妙度为3 + 2 = 5
音符2 ~ 3，美妙度为2 + (-6) = -4
音符3 ~ 4，美妙度为(-6) + 8 = 2
音符1 ~ 3，美妙度为3 + 2 + (-6) = -1
音符2 ~ 4，美妙度为2 + (-6) + 8 = 4
最优方案为：乐曲由和弦1，和弦3，和弦5组成，美妙度为5 + 2 + 4 = 11。

 

对于每个i，可能作为序列的左端点可以用【l,r】表示，

对于每个三元组【i,l,r】都存在一个最大值s，这可以用st表维护

而存在很多的【i,l,r】的最大值，可以用堆来维护

每取出一个最大值，【i,l,r】可以分裂成【i,l,mid-1】和【i,mid+1,r】（注意区间中至少有一个元素）

 

注：因为ans=s[i]-s[mid-1],所以查找最小值的区间为【l-1,r-1】,mid=MAX【l-1,r-1】+1;

然而如果你照我这么写，可能会掉成80分，因为l-1——r-1左端点可能为0，所以st表还要包括0

这样就ac了

#include<cstdio>
#include<iostream>
#include<queue>
using namespace std;
inline int read()
{
    int ret=0; bool f=0;
    char ch=getchar(); 
    while(ch<'0'||ch>'9') 
    {
        if(ch=='-') f=1;
        ch=getchar();
    }
    while(ch>='0'&&ch<='9')
        ret=(ret<<1)+(ret<<3)+ch-'0',ch=getchar();
    return f?-ret:ret;
}
 
int n,k,L,R,l,r;
long long ans;
const int N=5e5+5;
int lo[N][21],lg[N];
long long f[N][21],s[N];
struct node{
    int i,l,r,mid;
};
priority_queue< node >q;
 
inline int get(int l,int r)
{
    int k=lg[r-l+1];
    if(f[l][k]>f[r-(1<<k)+1][k])
        return lo[r-(1<<k)+1][k]; 
            else return lo[l][k];
}
 
bool operator <(node x,node y)
{
    return s[x.i]-s[x.mid-1]<s[y.i]-s[y.mid-1];
}
 
bool operator >(node x,node y)
{
    return s[x.i]-s[x.mid-1]>s[y.i]-s[y.mid-1];
}
int main()
{
    n=read(),k=read();
    L=read(),R=read();  
     
    f[0][0]=lo[0][0]=0;
    for(int i=1;i<=n;i++)
        f[i][0]=s[i]=s[i-1]+read(),
        lo[i][0]=i;
    
    lg[0]=-1;
    for(int i=1;i<=n+1;i++)
    	lg[i]=lg[i>>1]+1;
    
    for(int j=1;j<=lg[n+1];j++)
        for(int i=0;i<=n-(1<<j)+1;i++)
            if(f[i][j-1]>f[i+(1<<(j-1))][j-1])
                f[i][j]=f[i+(1<<(j-1))][j-1],
                lo[i][j]=lo[i+(1<<(j-1))][j-1];
                    else   
                        f[i][j]=f[i][j-1],
                        lo[i][j]=lo[i][j-1];
     
    for(int i=L;i<=n;i++)
        l=max(i-R+1,1),r=i-L+1,
        q.push((node){i,l,r,get(l-1,r-1)+1});
        
    ans=0;
    while(k--)
    {
        int i=q.top().i,mid=q.top().mid;
        l=q.top().l,r=q.top().r;
        ans+=s[i]-s[mid-1];
        q.pop();
        if(mid-1>=l)q.push((node){i,l,mid-1,get(l-1,mid-2)+1});
        if(mid+1<=r)q.push((node){i,mid+1,r,get(mid,r-1)+1});
    }
     
    printf("%lld\n",ans);
    return 0;
}