3747: [POI2015]Kinoman

题目链接

题目大意：给定f[]，g[]数组，在f[]数组中选择一个区间[l,r]，要求最大化$f_{l,r}$中只出现一次的数的权值

题解：经典套路

枚举区间左端点，线段树维护每个位置作为右端点的答案，取max

首先预处理出l=1的情况，然后每次l+1时，
$[l,next[l]-1]$的答案减少$w[f[l]]$
$[next[l],next[next[l]]-1]$的答案增加$w[f[l]]$

$r<l$的答案不需要特殊处理

我的收获：套路++

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

#define ll long long
#define INF 0x3f3f3f3f
#define M 1000005
#define ls x<<1
#define rs x<<1|1
#define lson l,mid,x<<1 
#define rson mid+1,r,x<<1|1
#define root 1,n,1

int n,m,f[M],w[M];
int next[M],last[M];
ll add[M<<2],mx[M<<2],ans;
bool mark[M];

void pushup(int x){mx[x]=max(mx[ls],mx[rs]);}

void pushdown(int x,int m)
{
    if(!add[x]) return ;
    add[ls]+=add[x],add[rs]+=add[x];
    mx[ls]+=add[x],mx[rs]+=add[x];
    add[x]=0;
}

void updata(int L,int R,int v,int l,int r,int x)
{
    if(L<=l&&r<=R){add[x]+=v,mx[x]+=v;return ;}
    pushdown(x,r-l+1);
    int mid=(l+r)>>1;
    if(L<=mid) updata(L,R,v,lson);
    if(R>mid) updata(L,R,v,rson);
    pushup(x);
}

void work()
{
    for(int i=1;i<=n;i++)
    {
        ans=max(ans,mx[1]);
        int t=next[i];
        if(t){
            updata(i,t-1,-w[f[i]],root);
            updata(t,!next[t]?n:next[t]-1,w[f[i]],root);
        }
        else updata(i,n,-w[f[i]],root);
    }
    printf("%lld\n",ans);
}

void init()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++) scanf("%d",&f[i]);
    for(int i=1;i<=m;i++) scanf("%d",&w[i]);
    for(int i=n;i>=1;i--) next[i]=last[f[i]],last[f[i]]=i;
    for(int i=1;i<=m;i++) if(last[i]) updata(last[i],!next[last[i]]?n:next[last[i]]-1,w[i],root);
}

int main()
{
    init();
    work();
    return 0;
}