bzoj 2006 [NOI2010]超级钢琴 rmq+堆

2006: [NOI2010]超级钢琴

Time Limit: 20 Sec  Memory Limit: 512 MB
Submit: 3708  Solved: 1846
[Submit][Status][Discuss]

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。

HINT

 

用三元组(i, l, r)表示右端点为i，左端点在[l, r]之间和最大的区间（[l, r]保证是对于i可行右端点区间的一个子区间），我们用堆维护一些这样的三元组。堆中初始的元素为每个i，并且[l, r]为这个i可行左端点的区间。假如某次最大值为(i, l, r)，并且j为那个和最大区间的左端点，那么需要往堆中加入两个三元组(i, l, j-1)和(i, j+1, r)。对于一个三元组，计算对应最大和的问题实际就是一个RMQ问题，可以通过Sparse Table在O(NlogN) – O(1)的时间内解决。这个算法的总时间复杂度为O(NlogN + K log(N+K))

 

#include<cstring>
#include<cstdio>
#include<algorithm>
#include<iostream>
#include<cmath>
#include<queue>

#define N 500007
#define ll long long
#define make(a,b,c,d) (Node){a,b,c,d}
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch)){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
    return x*f;
}

int n,K,L,R;
int a[N],mx[N][22];
ll ans;
int bin[22],Log[N];
struct Node
{
    int i,l,r,t;
    friend bool operator<(Node x,Node y)
    {
        return a[x.t]-a[x.i-1]<a[y.t]-a[y.i-1];
    }
};

void init()
{
    Log[0]=-1;for (int i=1;i<=n;i++) Log[i]=Log[i>>1]+1;
    for (int i=1;i<=n;i++) mx[i][0]=i;
    for (int i=n;i>=0;i--)
        for (int j=1;j<=18;j++)
            if (i+bin[j]-1<=n)
            {
                int t1=mx[i][j-1],t2=mx[i+bin[j-1]][j-1];
                mx[i][j]=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+1];
    int t1=mx[l][t],t2=mx[r-bin[t]+1][t];
    return a[t1]>a[t2]?t1:t2;
}
void solve()
{
    priority_queue<Node,vector<Node> >q;
    for (int i=1;i<=n;i++)
        if (i+L-1<=n)
        {
            int t=min(n,i+R-1);
            q.push(make(i,i+L-1,t,query(i+L-1,t)));
        }
    for (int i=1;i<=K;i++)
    {
        Node t=q.top();q.pop();
        ans+=a[t.t]-a[t.i-1];
        if (t.t-1>=t.l) q.push(make(t.i,t.l,t.t-1,query(t.l,t.t-1)));
        if (t.t+1<=t.r) q.push(make(t.i,t.t+1,t.r,query(t.t+1,t.r)));
    }
}
int main()
{
    bin[0]=1;for (int i=1;i<=20;i++) bin[i]=bin[i-1]<<1;
    n=read(),K=read(),L=read(),R=read();
    for (int i=1;i<=n;i++) a[i]=read();
    for (int i=1;i<=n;i++) a[i]+=a[i-1];
    init(),solve();
    printf("%lld\n",ans);
}