[BZOJ2006][NOI2010]超级钢琴(st表+堆贪心)

题目:

我是超链接

题解:

一个区间里的所有权值和要用前缀和相减表示啊,预处理出前缀和的区间最大值
目的O(1)查询给定范围内的最大值及其位置,固定一个前缀和最小的地方再去[l,r]区间里找一个前缀和最大的位置
我们就得到了以i为起点,[L,R]范围内的最优值及其位置,维护位置看起来很复杂,实际上只需要给st表加一个结构体就好了
把那些东西push进堆里去,弹出来第一个点时,ta的最优值的位置肯定要废掉,那就重新取最优值,为了避开maxlocation(loc),就把原来的区间分成两个区间[L,loc-1]和[loc+1,R],再找到最优位置和最优值push进去
选取k次就是答案了啊
不是所有的点左右边界都是[L,R]啊!!!区间还要分呢,还需要维护左右边界值呢

代码:

#include <queue>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <iostream>
#define LL long long 
#define sz 20
#define N 500005
using namespace std;
struct maxn{LL z;int loc;}maxx[N][sz];
struct hh{int id;LL z;int loc,lm,rm;};
bool operator <(const hh &a,const hh &b){return a.z<b.z;} 
priority_queue<hh>q;
int a[N],n;LL sum[N],ans;
void st()
{
    int i,j;
    for (j=1;j<sz;j++)
      for (i=1;i<=n;i++)
        if (i+(1<<j)-1<=n)
        {
            if (maxx[i][j-1].z<maxx[i+(1<<j-1)][j-1].z)
              maxx[i][j]=maxx[i+(1<<j-1)][j-1];
            else
              maxx[i][j]=maxx[i][j-1];
        }
}
int main()
{
    int kk,ll,rr,i,lm,rm;
    scanf("%d%d%d%d",&n,&kk,&ll,&rr);
    for (i=1;i<=n;i++) scanf("%d",&a[i]),sum[i]=sum[i-1]+a[i];
    for (i=1;i<=n;i++) maxx[i][0].z=sum[i],maxx[i][0].loc=i;
    st();
    for (i=1;i<=n;i++)
    {
        int l=i+ll-1,r=i+rr-1;
        lm=l;rm=min(r,n);r=rm;
        if (l>r) break;
        int k=log2(r-l+1),loc,Max;
        if (maxx[l][k].z<maxx[r-(1<<k)+1][k].z)
          loc=maxx[r-(1<<k)+1][k].loc,Max=maxx[r-(1<<k)+1][k].z;
        else loc=maxx[l][k].loc,Max=maxx[l][k].z;
        q.push((hh){i,Max-sum[i-1],loc,lm,rm});     
    }
    for (i=1;i<=kk;i++)
    {
        hh now=q.top(); q.pop();
        ans+=now.z;
        int k,loc,Max;
        int l=now.lm,r=now.loc-1;   
        if (l<=r)
        {
            k=log2(r-l+1);
            if (maxx[l][k].z<maxx[r-(1<<k)+1][k].z)
              loc=maxx[r-(1<<k)+1][k].loc,Max=maxx[r-(1<<k)+1][k].z;
            else loc=maxx[l][k].loc,Max=maxx[l][k].z;
            q.push((hh){now.id,Max-sum[now.id-1],loc,now.lm,r});
        }   
        l=now.loc+1,r=now.rm;
        if (l<=r)
        {
        k=log2(r-l+1);
        if (maxx[l][k].z<maxx[r-(1<<k)+1][k].z)
          loc=maxx[r-(1<<k)+1][k].loc,Max=maxx[r-(1<<k)+1][k].z;
        else loc=maxx[l][k].loc,Max=maxx[l][k].z;
        q.push((hh){now.id,Max-sum[now.id-1],loc,l,now.rm});    
        }
    }
    printf("%lld",ans);
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值