ST表+堆——BZOJ2006/Luogu2048 [NOI2010]超级钢琴

6 篇文章 0 订阅
1 篇文章 0 订阅

题面:BZOJ2006 Luogu2048
权当ST表练手第一题
题面大意是求出前K大连续字段和(长度在l到r之间)的和
首先想到贪心的想法,我们要找到以i为开头的长度在l到r之间的最大连续字段和,然后丢到堆里面,每次都取最大的就是答案了
但是可能以i为开头的对答案有贡献的方案有很多种,所以我们还要进一步做,考虑把这段区间分开
我们首先记下原序列最大时结尾的下标,然后分开的时候直接下标左边右边算,再扔进堆里就好了
贪心想法弄好了,现在问题是怎么求这个最大连续字段和
其实我们可以直接RMQ求解,把所有读入序列做前缀和,然后在l到r区间(长度)找到最大的那个,减掉开头的就是了
RMQ很多东西都可以求,线段树是log的(这个好像会T我也不清楚),但是ST表可以做到O(1)
所以直接ST表维护RMQ了

#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <iostream>
#include <ctime>
#include <map>
#include <queue>
#include <cstdlib>
#include <string>
#include <climits>
#include <set>
#include <vector>
using namespace std;
struct rppr{int v,w;}f[500001][21];
struct ppap{int v,i,l,r,w;};
bool operator <(ppap a,ppap b){return a.v<b.v;}
priority_queue<ppap>q;
long long ans=0;
int a[500001],n,K,L,R;
inline void pre(){
    for(int j=1;j<=20;j++)
        for(int i=1;i+(1<<j)-1<=n;i++)
            f[i][j]=(f[i][j-1].v>f[i+(1<<j>>1)][j-1].v)?f[i][j-1]:f[i+(1<<j>>1)][j-1];
}
inline rppr smax(int l,int r){
    int t=(int)(log(r-l+1.0)/log(2.0));
    return (f[l][t].v>f[r-(1<<t)+1][t].v)?f[l][t]:f[r-(1<<t)+1][t];
}//上面两个函数ST表
int main()
{
    scanf("%d%d%d%d",&n,&K,&L,&R);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);a[i]+=a[i-1];
        f[i][0]=(rppr){a[i],i};//前缀和+预处理
    }
    pre();rppr rp;ppap aa;
    for(int i=1;i<=n-L+1;i++){
        int l=i+L-1,r=min(i+R-1,n);
        rp=smax(l,r);
        q.push((ppap){rp.v-a[i-1],i,l,r,rp.w});
    }
    while(K--){
        aa=q.top();q.pop();
        ans+=(long long)aa.v;
        if(aa.w-1>=aa.l){
            rp=smax(aa.l,aa.w-1);
            q.push((ppap){rp.v-a[aa.i-1],aa.i,aa.l,aa.w-1,rp.w});
        }
        if(aa.w+1<=aa.r){
            rp=smax(aa.w+1,aa.r);
            q.push((ppap){rp.v-a[aa.i-1],aa.i,aa.w+1,aa.r,rp.w});
        }//区间分开
    }
    printf("%lld",ans);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值