P2048 超级钢琴(RMQ+优先队列)

题目描述

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

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

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

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

输入格式

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

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

输出格式

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

输入输出样例

输入 #1复制

4 3 2 3
3
2
-6
8

输出 #1复制

11

说明/提示

共有5种不同的超级和弦：

1.    音符1 ~ 2，美妙度为3 + 2 = 5
2.    音符2 ~ 3，美妙度为2 + (-6) = -4
3.    音符3 ~ 4，美妙度为(-6) + 8 = 2
4.    音符1 ~ 3，美妙度为3 + 2 + (-6) = -1
5.    音符2 ~ 4，美妙度为2 + (-6) + 8 = 4

最优方案为：乐曲由和弦1，和弦3，和弦5组成，美妙度为5 + 2 + 4 = 11。

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

解题思路：

第一种做法是先求出前缀和。

用ST表预处理出从[l,r]的前缀和最大值。

然后有一个结构体，struct{ pos, l, r, val},各个元素的意思是：从pos这个位置到[l,r]这个区间的各个区间的和最大，这个val可以根据ST表求出来，把这些值先放进优先队列，为val重载小于运算符。然后不断从优先队列中取出大的值，再将取完后的结构体分解成两个结构体存到优先队列中。

#include <iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#include<cmath>
#include<algorithm>

using namespace std;
const int maxn = 5e5+5;

int st[maxn][20];
int a[maxn], sum[maxn];

void init(int n)
{
    int N = floor(log2(n));
    for(int j = 1; j <= N; j++)
    {
        for(int i = 1; i <= n+1-(1<<j); i++)
        {
            if(sum[st[i][j-1]]>sum[st[i+(1<<(j-1))][j-1]])
                st[i][j] = st[i][j-1];
            else
                st[i][j] = st[i+(1<<(j-1))][j-1];
        }
    }
}
int get(int l, int r)
{
    int k = floor(log2(r+1-l));
    if(sum[st[l][k]]>sum[st[r+1-(1<<k)][k]])
        return st[l][k];
    else
        return st[r+1-(1<<k)][k];
}

struct node{
    int pos, l, r,pp,val;
    int operator <(const node &rhs) const{
        return val < rhs.val;
    }
};
priority_queue<node>q;

int main()
{
    int n, k, l, r;
    scanf("%d%d%d%d", &n, &k, &l, &r);
    l--;r--;
    for(int i = 1; i <= n; i++)
    {
        scanf("%d", &a[i]);
        sum[i] = sum[i-1]+a[i];
        st[i][0] = i;
    }
    init(n);
//    cout << sum[get(3,4)]-sum[0]<<endl;
    int pos;
    for(int i = 1; i+l <= n; i++)
    {
        int ll = i+l;
        int rr = min(i+r,n);
        pos = get(ll,rr);
        q.push(node{i,ll,rr,pos,sum[pos]-sum[i-1]});
    }
    long long ans = 0;
    while(k--)
    {
        node no =q.top();
        ans+=no.val;
        q.pop();
        if(no.pp-1>=no.l){
            pos = get(no.l,no.pp-1);
            q.push(node{no.pos,no.l,no.pp-1,pos,sum[pos]-sum[no.pos-1]});
        }
        if(no.pp+1<=no.r)
        {
            pos = get(no.pp+1,no.r);
            q.push(node{no.pos,no.pp+1,no.r,pos,sum[pos]-sum[no.pos-1]});
        }
    }
    printf("%lld\n", ans);
    return 0;
}