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
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;
}

 

 

 

 

 

你可能感兴趣的:(模板类(什么,这也是模板),思维(抖啊,抖啊,抖个机灵儿))