BZOJ2006:[NOI2010]超级钢琴

2006: [NOI2010]超级钢琴

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

 

思路{

　　贪心的想，我们一定是按照贡献值排名从大到小取得收益。

　　直观的思路是把每段序列丢到一个堆里面就可以了。

　　但是数据范围是明摆着会T的。考虑优化这个加入的过程。

　　排名，考虑排名，那我们就可以固定一个起点来考虑。

　　一段连续的区间长度[L,R]对应了一堆选择。那固定起点的话，

　　对于这个起点排名为k的位置,那在前缀和中他的前缀和排名也为k吧。

　　那么我们就可以把序列拆成2部分，sum[R]-sum[L-1]，那我就可以用主席树加速这个过程;

　　按照排名丢入，每次取堆顶直接累加答案，将当前答案的所在端点的下一个排名的贡献加入堆。

　　这样就加速了过程。

}

#include<bits/stdc++.h>
#define RG register
#define il inline
#define LL long long
#define N 20000005
using namespace std;
int rs[N],ls[N],rt[N],tree[N],n,k,L,R,aa;LL sum[500001],id;
#define mid ((l+r)>>1)
void Insert(int y,int & x,LL l,LL r,LL val){
  x=++id;tree[x]=tree[y]+1;rs[x]=rs[y],ls[x]=ls[y];if(l==r)return;
  if(mid<val)Insert(rs[y],rs[x],mid+1,r,val);
  else Insert(ls[y],ls[x],l,mid,val);
}
LL Query(int y,int x,LL l,LL r,LL val){
  if(l==r)return l;int tmp=tree[ls[x]]-tree[ls[y]];
  if(tree[x]-tree[y]<val)return 1e9;
  if(tmp<val)return Query(rs[y],rs[x],mid+1,r,val-tmp);
  else return Query(ls[y],ls[x],l,mid,val);
}
struct node{
  int pos,rnk;LL val;
  bool operator < (const node & a) const{return val<a.val;}
};LL Ans;
#define lll -500001000
#define rrr 500001000
priority_queue<node>hp;
int main(){
  scanf("%d%d%d%d",&n,&k,&L,&R);Insert(0,rt[0],lll,rrr,0);
  for(int i=1;i<=n;++i)scanf("%d",&aa),sum[i]=sum[i-1]+aa;
  for(int i=1;i<=n;++i)Insert(rt[i-1],rt[i],lll,rrr,sum[i]);
  for(int l=L;l<=n;++l){int ri=l-L,le=l-R-1;
    LL x;if(le<0)x=Query(0,rt[ri],lll,rrr,1);
    else x=Query(rt[le],rt[ri],lll,rrr,1);
    hp.push((node){l,1,sum[l]-x});
  }while(k--){
    node tmp=hp.top();hp.pop();
    int l=tmp.pos,ri=l-L,le=l-R-1,rnk=tmp.rnk;
    LL x;if(le<0)x=Query(0,rt[ri],lll,rrr,rnk+1);
    else x=Query(rt[le],rt[ri],lll,rrr,rnk+1);Ans+=tmp.val;
    hp.push((node){l,rnk+1,sum[l]-x});
  }printf("%lld\n",Ans);
  return 0;
}