bzoj2006 NOI2010 数据结构+堆维护区间和最大

2006: [NOI2010]超级钢琴

Time Limit: 20 Sec  Memory Limit: 552 MB
Submit: 3431  Solved: 1686
[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。

有点不好说明题解，，，上代码算了

其实就是对于每个点，寻找以他为开头长为[l,r]的序列的权值和最大的，找完最大的找第二大的，然后第三大，以此类推

方法很巧妙

i维护开头，j维护最大点位置，l，r维护选择区间，val维护区间总值

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#include<queue>
#define inf 0x3f3f3f3f
#define ll long long
#define N 500005
using namespace std;
int n,m,lx,rx,st[N][],pos[N][],sum[N],a[N],lg[N];
struct node{
    int i,j,l,r,val;
    bool operator < (const node &b)const{
        return val==b.val?i>b.i:val<b.val;
    }
};
char gc(){
    static char s[],*p1,*p2;
    if(p1==p2)p2=(p1=s)+fread(s,,,stdin);
    if(p1==p2)return EOF;
    return *p1++;
}

void r(int &x){
    x=;int f=;char ch=gc();
    while(ch>''||ch<''){
        if(ch=='-')f=-;
        ch=gc();
    }
    while(ch>=''&&ch<='')x=x*+ch-'',ch=gc();
    x*=f;
}
priority_queue<node>q;
void pre(){
    for(register int i=;i<=n;i++)
    pos[i][]=i,st[i][]=sum[i];
    for(register int j=;j<=;j++)
    for(register int i=;i+(<<j)-<=n;i++){
        if(st[i][j-]>st[i+(<<j-)][j-])
        st[i][j]=st[i][j-],pos[i][j]=pos[i][j-];
        else st[i][j]=st[i+(<<j-)][j-],pos[i][j]=pos[i+(<<j-)][j-];
    }
}
int qmx(int l,int r){
    int t=lg[r-l+];
    return max(st[l][t],st[r-(<<t)+][t]);
}
int qos(int l,int r){
    int t=lg[r-l+];
    if(st[l][t]>st[r-(<<t)+][t])return pos[l][t];
    return pos[r-(<<t)+][t];
}
int main(){
#ifdef wsy
    freopen("data.in","r",stdin);
#else
    freopen("piano.in","r",stdin);
    freopen("piano.out","w",stdout);
#endif
    r(n);r(m);r(lx);r(rx);
    for(register int i=;i<=n;i++)
    r(a[i]),sum[i]=sum[i-]+a[i];
    pre();for(register int i=;i<=n;i++)lg[i]=lg[i>>]+;
    for(register int i=;i<=n;i++){
        int l=i+lx-,r=min(i+rx-,n);
        if(l>n)break;
        int val=qmx(l,r)-sum[i-];
        int p=qos(l,r);
        q.push((node){i,p,l,r,val});
    }
    ll ans=;
    while(m--){
        node u=q.top();q.pop();
        ans+=u.val;
        if(u.j>u.l){
            int l=u.l,r=u.j-;
            int p=qos(l,r),val=qmx(l,r)-sum[u.i-];
            q.push((node){u.i,p,l,r,val});
        }
        if(u.j<u.r){
            int l=u.j+,r=u.r;
            int p=qos(l,r),val=qmx(l,r)-sum[u.i-];
            q.push((node){u.i,p,l,r,val});
        }
    }
    printf("%I64d",ans);
    return ;
}