[BZOJ2006][NOI2010]超级钢琴-ST表+堆

超级钢琴

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。

---

所以咱真的不知道这题用主席树的dalao们是怎么做的……
（虽然咱也没想出现在用的这个来自其他dalao的方法Orz）

---

思路:
考虑一个十分巧妙的算法。
首先先处理出前缀和。
定义四元组(i,l,r,j)表示当前四元组左端点为i，可行的右端点区间为[l,r]，在当前可行右端点区间中找到的前缀和的值最大的一个位置是j。
j的查询显然可以使用ST表、线段树之类的东西。
咱将n个这样的四元组插入一个堆，堆内按[i,j]区间值大小排序，这个显然可以用前缀和计算。
然后循环k次，每次取出堆顶的四元组，把取出四元组的值加入答案，然后分裂这个四元组成两个，把分裂出的两个四元组插回堆中。
然后咱就得到了答案。
复杂度 O(稳) O(nlogn+klog(n+k))

那么关键的问题来了:怎么分裂呢?
对于一个四元组(i,l,r,j)，只需要分成(i,l,j-1,新计算的j)和(i,j+1,r,新计算的j)即可。
因为咱只是单纯地删掉了j这个位置作为i的右端点的可能性罢了~

然后？没有然后了……

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

using namespace std;

typedef long long ll;

inline ll read()
{
    ll x=0,f=1;char ch=getchar();
    while(ch<'0' || '9'<ch){if(ch=='-')f=-1;ch=getchar();}
    while('0'<=ch && ch<='9')x=x*10+(ch^48),ch=getchar();
    return x*f;
}

const ll N=500009;

ll n,k,l,r,ans,a[N],st[N][26];

struct node
{
    ll i,l,r,j;
    bool operator < (node b)const
    {
        return  a[j]-a[i-1]<a[b.j]-a[b.i-1];
    }
};


priority_queue<node> q;

inline ll query(ll l,ll r)
{
    if(l==r)return l;
    ll pos=log2(r-l+1);
    if(a[st[l][pos]]>a[st[r-(1<<pos)+1][pos]])
        return st[l][pos];
    else
        return st[r-(1<<pos)+1][pos];
}

int main()
{
    n=read();k=read();
    l=read();r=read();

    for(ll i=1;i<=n;i++)
        a[i]=read()+a[i-1];

    for(ll i=1;i<=n;i++)
        st[i][0]=i;

    for(ll i=1;i<=18;i++)
        for(ll j=1;j+(1<<i)-1<=n;j++)
        {
            if(a[st[j][i-1]]>a[st[j+(1<<i-1)][i-1]])
                st[j][i]=st[j][i-1];
            else
                st[j][i]=st[j+(1<<i-1)][i-1];
        }

    for(ll i=1,e=n-l+1;i<=e;i++)
    {
        ll minr=min(i+r-1,n);
        q.push((node){i,i+l-1,minr,query(i+l-1,minr)});
    }

    for(ll i=1;i<=k;i++)
    {
        node tmp=q.top();q.pop();
        ans+=a[tmp.j]-a[tmp.i-1];
        if(tmp.l<=tmp.j-1)
            q.push((node){tmp.i,tmp.l,tmp.j-1,query(tmp.l,tmp.j-1)});
        if(tmp.j+1<=tmp.r)
            q.push((node){tmp.i,tmp.j+1,tmp.r,query(tmp.j+1,tmp.r)});
    }

    printf("%lld\n",ans);
    return 0;
}