BZOJ2006: [NOI2010]超级钢琴

2006: [NOI2010]超级钢琴

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

HINT

Source

【题解】

初始时，对于每一个右端点，他所选的最优左端点很容易求出：前缀和+st表

贪心选区间和最大的（堆实现）。选了某个右端点i，对应左端点pos后，

只要i不被选，以i为右端点的区间的左端点不是pos的区间就不可能被选

pos被选了之后

我们不难发现

右端点i的最优左端点在i - r + 1, pos - 1 和 pos + 1, i - l + 1里面

取两个左端点最优值压入堆中嘛？当然不是，两个左端点都应加入堆中。

两个端点一个能构成最大，另一个可能构成次大。

因long long和st表的ij弄反被卡

#include <iostream>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <queue>
#include <vector>
#define max(a, b) ((a) > (b) ? (a) : (b))
#define min(a, b) ((a) < (b) ? (a) : (b))

inline void read(long long &x)
{
    x = 0;char ch = getchar(), c = ch;
    while(ch < '0' || ch > '9')c = ch, ch = getchar();
    while(ch <= '9' && ch >= '0')x = x * 10 + ch - '0', ch = getchar();
    if(c == '-')x = -x;
}

const long long MAXN = 500000 + 10;
const long long MAXK = 500000 + 10;

long long n,k,l,r,num[MAXN],st[20][MAXN],rank[20][MAXN],M,pow2[30],ans;

struct Node
{
    long long value, l, r, rank, i;
    Node(long long _value, long long _l, long long _r, long long _rank, long long _i){value = _value, l = _l, r = _r,rank = _rank;i = _i;}    
    Node(){}
};

struct cmp
{
    bool operator()(Node a, Node b)    
    {
        return a.value < b.value;
    }
};

std::priority_queue<Node, std::vector<Node>, cmp> q;

void yuchuli()
{
    while(pow2[M] <= n)++ M;
    --M;
    for(register long long i = 1;i <= M;++ i)
        for(register long long j = 0;j <= n;++ j)
        {
            if(j + pow2[i] - 1 > n)continue;
            if(st[i - 1][j] < st[i - 1][j + pow2[i - 1]])
            {
                st[i][j] = st[i - 1][j];
                rank[i][j] = rank[i - 1][j];
            }
            else
            {
                st[i][j] = st[i - 1][j + pow2[i - 1]];
                rank[i][j] = rank[i - 1][j + pow2[i - 1]];
            }
        }
}

long long find(long long l, long long r)
{
    if(l > r)
    {
        long long tmp = l;
        l = r;
        r = tmp;
    }
    long long m = 0;
    while(pow2[m] <= (r - l + 1))++ m;
    -- m;
    if(st[m][l] < st[m][r - pow2[m] + 1])
        return rank[m][l];
    else
        return rank[m][r - pow2[m] + 1];
}

int main()
{
    read(n);read(k);read(l);read(r);
    pow2[0] = 1;
    for(register long long i = 1;i <= 25;++ i)pow2[i] = (pow2[i - 1] << 1);
    for(register long long i = 1;i <= n;++ i)
    {
        read(num[i]);
        num[i] += num[i - 1];
        st[0][i] = num[i];
        rank[0][i] = i;
    }
    yuchuli();
    for(register long long i = l;i <= n;++ i)
    {
        long long k = find(max(0, i - r), max(i - l, 0));
        q.push(Node(num[i] - num[k], max(0, i - r), max(i - l, 0), k, i));
    }
    long long now = 0;
    register Node tmp;
    register long long pos;
    for(;now < k;++ now)
    {
        tmp = q.top(), q.pop();
        long long l = tmp.l, r = tmp.r;
        ans += tmp.value; 
        if(tmp.l <= tmp.rank - 1 && tmp.l >= 0)
        {
            pos = find(max(0, tmp.l), max(tmp.rank - 1, 0));
            q.push(Node(num[tmp.i] - num[pos], tmp.l, tmp.rank - 1, pos, tmp.i));
        }
        if(tmp.r >= tmp.rank + 1)
        {
            pos = find(max(0, tmp.rank + 1), max(tmp.r, 0));
            q.push(Node(num[tmp.i] - num[pos], tmp.rank +1, tmp.r, pos, tmp.i));
        }
    }
    printf("%lld", ans);
    return 0;
}