洛谷 P2048 [NOI2010]超级钢琴

题面

题意

给出一个序列,求和最大的k个连续子序列的和,且连续子序列长度在l与r之间.

做法

首先预处理出前缀和,我们可以从左端点开始考虑,若左端点为i,则右端点在i+l-1与i+r-1之间,那么可以计算出这些区间的最大值,然后放到堆里去,每次从堆中取出最大值加到ans中后再将该区间左右两区间的最大值放入堆中.
也就是说我们维护一个结构体(i,l,r,mx)表示左端点为i,右端点在l,r之间的,且这些区间中最大的一个区间的右端点为mx,那么当这个结构体从堆中弹出来后,我们可以将结构体(i,l,mx-1,max(l~mx-1))和结构体(i,mx+1,r,max(mx+1~r))放到堆里,利用st表,我们可以快速求出max(l~mx-1)和max(mx+1~r),重复k次计算最大值的和即可.

代码

#include<iostream>
#include<cstdio>
#include<queue>
#define ll long long
#define N 500100
using namespace std;

ll n,m,l,r,num[N],qz[N],st[20][N],pos[20][N],ans,lg[N];
struct Node
{
    ll from,left,right,mx_pos;
    bool operator < (const Node &u) const
    {
        return qz[mx_pos]-qz[from-1]<qz[u.mx_pos]-qz[u.from-1];
    }
}node[N],tmp,t2;
priority_queue<Node>pq;

inline ll lo(ll u)
{
    ll res=0;
    for(;u;u>>=1,res++);
    return res-1;
}

inline ll ask(ll u,ll v)
{
    ll l=lg[v-u];
    if(st[l][u]<st[l][v-(1 << l)+1])
        return pos[l][v-(1 << l)+1];
    return pos[l][u];
}

int main()
{
    ll i,j;
    cin>>n>>m>>l>>r;
    for(i=1;i<=n;i++)
    {
        scanf("%lld",&num[i]);
        qz[i]=qz[i-1]+num[i];
        st[0][i]=qz[i];
        pos[0][i]=i;
        lg[i]=lo(i);
    }
    for(i=1;(1 << i)<=n;i++)
    {
        for(j=1;j+(1 << i)-1<=n;j++)
        {
            if(st[i-1][j]>st[i-1][j+(1 << (i-1))])
            {
                st[i][j]=st[i-1][j];
                pos[i][j]=pos[i-1][j];
            }
            else
            {
                st[i][j]=st[i-1][j+(1 << (i-1))];
                pos[i][j]=pos[i-1][j+(1 << (i-1))];
            }
        }
    }
    for(i=1;i+l-1<=n;i++)
    {
        node[i].from=i;
        node[i].left=i+l-1;
        node[i].right=min(n,i+r-1);
        node[i].mx_pos=ask(node[i].left,node[i].right);
        pq.push(node[i]);
    }
    for(i=1;i<=m;i++)
    {
        tmp=pq.top();
        pq.pop();
        ans+=qz[tmp.mx_pos]-qz[tmp.from-1];
        t2=tmp;
        t2.right=t2.mx_pos-1;
        if(t2.left<=t2.right)
        {
            t2.mx_pos=ask(t2.left,t2.right);
            pq.push(t2);
        }
        t2=tmp;
        t2.left=t2.mx_pos+1;
        if(t2.left<=t2.right)
        {
            t2.mx_pos=ask(t2.left,t2.right);
            pq.push(t2);
        }
    }
    cout<<ans;
}