Kinoman线段树

共有m部电影,编号为1~m,第i部电影的好看值为w[i]。
在n天之中(从1~n编号)每天会放映一部电影,第i天放映的是第f[i]部。
你可以选择l,r(1<=l<=r<=n),并观看第l,l+1,…,r天内所有的电影。如果同一部电影你观看多于一次,你会感到无聊,于是无法获得这部电影的好看值。所以你希望最大化观看且仅观看过一次的电影的好看值的总和。

 

第一行两个整数n,m(1<=m<=n<=1000000)。
第二行包含n个整数f[1],f[2],…,f[n](1<=f[i]<=m)。
第三行包含m个整数w[1],w[2],…,w[m](1<=w[j]<=1000000)。

输出观看且仅观看过一次的电影的好看值的总和的最大值。

 

这里考虑每一个电影对某一个区间的影响,用线段树维护一个区间最大值

pre[i]为第i个电影前一次出现的位置,a[i]为第i个电影的编号,那么w[a[i]]就是第i个电影的好看值

那么第i个电影对[pre[i]+1,i]这个区间显然有w[a[i]]的贡献,那么用线段树区间更新[pre[i]+1,i]

同时第i个电影,对之前更新过的区间[pre[pre[i]]+1,pre[i]]有着一个-w[a[i]]的贡献

(即消除之前的影响,因为每次查询区间最大值查询的是[1,i],即[1,i]中最大的区间最大值

,因为[1,i]中所有的电影都处理过了,所以可以查询)

注意这里当之前更新过(pre[i]不为0)的时候才能对之前的区间进行取消更新操作,

否则就相当于对[1,pre[i]]pre[pre[i]]=0)进行了取消更新操作,这就错了

还有,注意懒标记要开longlong,因为范围超过int了

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int maxn = 1e6+7;
ll a[maxn],n,m,pre[maxn],w[maxn],last[maxn];
struct node{
    int l,r;
    ll Maxsum,lazy;
    void updata(ll val){
        Maxsum += val;
        lazy += val;
    }
}tree[maxn<<2];
void push_up(int x){
    tree[x].Maxsum = max(tree[x<<1].Maxsum, tree[x<<1|1].Maxsum);
}
void push_down(int x){
    ll lazy = tree[x].lazy;
    if(lazy){
        tree[x].lazy = 0;
        tree[x<<1].updata(lazy);
        tree[x<<1|1].updata(lazy);
    }
}
void build(int x,int l,int r){
    tree[x].l = l,tree[x].r = r;
    tree[x].lazy = tree[x].Maxsum = 0;
    if(l != r){
        int mid = (l+r)>>1;
        build(x<<1,l,mid);
        build(x<<1|1,mid+1,r);
        push_up(x);
    }
}
void updata(int x,int l,int r,ll val){
    int L = tree[x].l,R = tree[x].r;
    if(l<=L && R<=r)
        tree[x].updata(val);
    else{
        int mid = (L+R)>>1;
        push_down(x);
        if(l<=mid) updata(x<<1,l,r,val);
        if(r>mid) updata(x<<1|1,l,r,val);
        push_up(x);
    }
}
ll query(int x,int l,int r){
    int L = tree[x].l,R = tree[x].r;
    if(l<=L && R<=r)
        return tree[x].Maxsum;
    else{
        int mid = (L+R)>>1;
        push_down(x);
        ll ans = 0;
        if(l<=mid) ans = max(ans, query(x<<1,l,r));
        if(r>mid) ans = max(ans, query(x<<1|1,l,r));
        push_up(x);
        return ans;
    }
}
int main(){
    scanf("%lld%lld",&n,&m);
    for(int i=1;i<=n;i++)
        scanf("%lld",&a[i]);
    for(int i=1;i<=m;i++)
        scanf("%lld",&w[i]);
    for(int i=1;i<=n;i++){
        pre[i] = last[a[i]];
        last[a[i]] = i;
    }
    build(1,1,n);
    ll ans = 0;
    for(int i=1;i<=n;i++){
        updata(1,pre[i]+1,i,w[a[i]]);
        if(pre[i]) updata(1,pre[pre[i]]+1,pre[i],-w[a[i]]);
        ans = max(ans,query(1,1,i));
    }
    printf("%lld\n",ans);
    return 0;
}

 

 

 

 

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值