超级钢琴（codevs 2934）

Posted 2020-08-08 Cola

题目描述 Description

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

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

一个“超级和弦”由若干个编号连续的音符组成，包含的音符个数不少于L且不多于R。我们定义超级和弦的美妙度为其包含的所有音符的美妙度之和。两个超级和弦被认为是相同的，当且仅当这两个超级和弦所包含的音符集合是相同的。

小Z决定创作一首由k个超级和弦组成的乐曲，为了使得乐曲更加动听，小Z要求该乐曲由k个不同的超级和弦组成。我们定义一首乐曲的美妙度为其所包含的所有超级和弦的美妙度之和。小Z想知道他能够创作出来的乐曲美妙度最大值是多少。

输入描述 Input Description

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

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

输出描述 Output Description

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

样例输入 Sample Input

4 3 2 3

3

2

-6

8

样例输出 Sample Output

11

数据范围及提示 Data Size & Hint

0<N<=500000,0<k<=50000

所有数据满足：-1000 ≤ Ai ≤ 1000，1 ≤ L ≤ R ≤ n且保证一定存在满足要求的乐曲。

/*
  可持续性线段树+堆
*/
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
#define N 500010
#define M 10000010
using namespace std;
int a[N],co[N],root[N],used[N],n,m,l,r,cnt;
struct node
{
    int sum,lc,rc;
};node t[M*4];
priority_queue< pair<int,int> > q;
int read()
{
    char c=getchar();int num=0,flag=1;
    while(c<\'0\'||c>\'9\'){if(c==\'-\')flag=-1;c=getchar();}
    while(c>=\'0\'&&c<=\'9\'){num=num*10+c-\'0\';c=getchar();}
    return num*flag;
}
int build(int v,int x,int y)
{
    int k=++cnt;t[k].sum=v;
    t[k].lc=x;t[k].rc=y;
    return k;
}
void insert(int &root,int pre,int l,int r,int pos)
{
    root=build(t[pre].sum+1,t[pre].lc,t[pre].rc);
    if(l==r)return;
    int mid=(l+r)/2;
    if(pos<=mid)insert(t[root].lc,t[pre].lc,l,mid,pos);
    else insert(t[root].rc,t[pre].rc,mid+1,r,pos);
}
int query(int x,int y,int l,int r,int k)
{
    if(l==r)return l;
    int mid=(l+r)/2;
    int sum=t[t[y].lc].sum-t[t[x].lc].sum;
    if(k<=sum)return query(t[x].lc,t[y].lc,l,mid,k);
    else return query(t[x].rc,t[y].rc,mid+1,r,k-sum);
}
int main()
{
    n=read();m=read();l=read();r=read();
    for(int i=1;i<=n;i++)
    {
        int x=read();
        a[i]=a[i-1]+x;
        co[i]=a[i];
    }
    sort(co+1,co+n+1);
    int num=unique(co+1,co+n+1)-co-1;
    for(int i=1;i<=n;i++)
    {
        int pos=lower_bound(co+1,co+num+1,a[i])-co;
        insert(root[i],root[i-1],1,num,pos);
    }
    for(int i=1;i+l-1<=n;i++)
    {
        int L=i+l-1,R=min(i+r-1,n);
        int len=(R-L+1);++used[i];
        int pos=query(root[L-1],root[R],1,num,len);
        q.push(make_pair(co[pos]-a[i-1],i));
    }
    long long ans=0;
    for(int i=1;i<=m;i++)
    {
        int p=q.top().second;
        ans+=q.top().first;q.pop();
        int L=p+l-1,R=min(p+r-1,n);
        int len=R-L+1;used[p]++;
        if(used[p]>=len+1)continue;
        int pos=query(root[L-1],root[R],1,num,len-used[p]+1);
        q.push(make_pair(co[pos]-a[p-1],p));
    }
    cout<<ans;
    return 0;
}

View Code