[題解]POI2015 Kinoman

Description

共有m部電影，編號爲1~m，第i部電影的好看值爲w[i]。在n天之中（從1~n編號）每天會放映一部電影，第i天放映的是第f[i]部。你可以選擇l,r(1<=l<=r<=n)，並觀看第l,l+1,…,r天內所有的電影。如果同一部電影你觀看多於一次，你會感到無聊，於是無法獲得這部電影的好看值。所以你希望最大化觀看且僅觀看過一次的電影的好看值的總和。
輸入輸出格式

Input

第一行兩個整數n,m(1<=m<=n<=1000000)。第二行包含n個整數f[1],f[2],…,fn。第三行包含m個整數w[1],w[2],…,wm。

Output

輸出觀看且僅觀看過一次的電影的好看值的總和的最大值。

Solution

預處理出下一個於某天放同樣電影的日子，然後從右往左枚舉L，線段樹中i位置存的數表示選當前L到i位置的所有電影的貢獻，然後區間加減來維護即可

代碼:

#include<cstdio>
#include<algorithm>
using namespace std;

template<typename T>inline void read(T &x){
    T f=1;char ch=getchar();
    for(;ch<'0'||ch>'9';ch=getchar())if(ch=='-')f=-1;
    for(x=0;ch>='0'&&ch<='9';ch=getchar())x=x*10+ch-'0';
    x*=f;
}

typedef long long LL;
const int maxn=1000010;
struct Segment_Tree{
    #define lc x<<1
    #define rc x<<1|1
    int L[maxn<<2],R[maxn<<2];
    LL mx[maxn<<2],add[maxn<<2];
    void Build(int x,int l,int r){
        mx[x]=add[x]=0;
        if((L[x]=l)==(R[x]=r))return;
        int mid=(l+r)>>1;
        Build(lc,l,mid);Build(rc,mid+1,r);
    }
    void pushdown(int x){
        if(add[x]){
            mx[lc]+=add[x];
            add[lc]+=add[x];
            mx[rc]+=add[x];
            add[rc]+=add[x];
            add[x]=0;
        }
    }
    void Add(int x,int l,int r,LL val){
        if(L[x]>=l&&R[x]<=r)return mx[x]+=val,add[x]+=val,void();
        pushdown(x);
        int mid=(L[x]+R[x])>>1;
        if(l<=mid)Add(lc,l,r,val);
        if(r>mid)Add(rc,l,r,val);
        mx[x]=max(mx[lc],mx[rc]);
    }
    LL Query(int x,int l,int r){
        if(L[x]>=l&&R[x]<=r)return mx[x];
        pushdown(x);
        int mid=(L[x]+R[x])>>1;LL ans=0;
        if(l<=mid)ans=max(ans,Query(lc,l,r));
        if(r>mid)ans=max(ans,Query(rc,l,r));
        return ans;
    }
}tree;
int n,m,f[maxn],w[maxn],suf[maxn],nxt[maxn];
LL ans;

int main(){
    read(n);read(m);
    for(int i=1;i<=n;i++)read(f[i]);
    for(int i=1;i<=m;i++)read(w[i]);
    for(int i=n;i>=1;i--){
        if(!nxt[f[i]])suf[i]=n+1;
        else suf[i]=nxt[f[i]];
        nxt[f[i]]=i;
    }
    suf[n+1]=n+1;
    tree.Build(1,1,n);
    for(int i=n;i>=1;i--){
        tree.Add(1,i,suf[i]-1,w[f[i]]);
        if(suf[i]<=n)tree.Add(1,suf[i],suf[suf[i]]-1,-w[f[i]]);
        ans=max(ans,tree.Query(1,i,n));
    }
    printf("%lld\n",ans);
    return 0;
}