BZOJ 2006: [NOI2010]超级钢琴

Time Limit: 20 Sec Memory Limit: 552 MB
Submit: 3254 Solved: 1604

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

Source

---

题解：

用st表处理出前缀和数组的区间最大值的数组下标，每次固定右端点之后我们就希望找到一个左边的最小值，然后把它用四元组存到堆里面去。
每次从堆里面弹出元素之后就把这个东西的左边找一个最大的和右边找一个最大的扔到堆里面去，然后反复贪心就可以了

---

#include<cstdio>
#include<cstring>
#include<iostream>
#include<utility>
#include<queue>
using namespace std;
const int MAXN=500000+10; 
int n,dp[MAXN][30],k,L,R,a[MAXN];
//int Log[MAXN];
priority_queue<pair<pair<int,int>,pair<int,int> > >q;
void get_st(){
    for(register int i=1;i<=n;i++) dp[i][0]=i;
    for(register int len=1;(1<<len)<=n;len++){
        for(register int i=0;i+(1<<len)-1<=n;i++){
            int j=i+(1<<len)-1;
            if(a[dp[i][len-1]]<a[dp[i+(1<<(len-1))][len-1]])dp[i][len]=dp[i][len-1];
            else                                            dp[i][len]=dp[i+(1<<(len-1))][len-1];
        }
    }
}
long long find(int L,int R){
    if(R<L) return -1;
    int M=0;
    while((1<<(M+1))<(R-L+1)) M++;
    //int M=Log[R-L+1];
    if(a[dp[L][M]]<a[dp[R-(1<<M)+1][M]])return dp[L][M];
    else                                return dp[R-(1<<M)+1][M];
}
int main(){
    scanf("%d%d%d%d",&n,&k,&L,&R);
    for(register int i=1;i<=n;i++) scanf("%lld",&a[i]),a[i]=a[i-1]+a[i];
    get_st();
    for(register int r=L;r<=n;r++){
        int l=max(0,r-R);
        int tmp=find(l,r-L);
        q.push(make_pair(make_pair(a[r]-a[tmp],r),make_pair(l,r-L)));
    }long long ans=0;
    for(register int i=1;i<=k;i++){
        pair<int,int>f=q.top().first;pair<int,int>s=q.top().second;
        ans+=f.first;
        int loc=f.second,lft=s.first,rgt=s.second;q.pop();
        int mid=find(lft,rgt);
        int loc1=find(lft,mid-1),loc2=find(mid+1,rgt);
        if(loc1!=-1) q.push(make_pair(make_pair(a[loc]-a[loc1],loc),make_pair(lft,mid-1)));
        if(loc2!=-1) q.push(make_pair(make_pair(a[loc]-a[loc2],loc),make_pair(mid+1,rgt)));
    }
    printf("%lld\n",ans);
    return 0;
}