Link
把贡献看做前缀和,第一次出现是\(+\),第二次出现是\(-\),之后的都是\(0\)。
那么预处理一下每个点后面一个和它同色的点是哪个,然后从右往左枚举左端点,线段树维护每个右端点的答案就好了。
#include<cstdio>
#include<cctype>
const int N=1000007;
using ll=long long;
int read(){int x=0,c=getchar();while(!isdigit(c))c=getchar();while(isdigit(c))x=x*10+c-48,c=getchar();return x;}
ll max(ll a,ll b){return a>b? a:b;}
int a[N],w[N],t[N],next[N];ll mx[N<<2],tag[N<<2];
#define ls p<<1
#define rs p<<1|1
#define mid ((l+r)>>1)
void pushup(int p){mx[p]=max(mx[ls],mx[rs]);}
void modify(int p,ll x){tag[p]+=x,mx[p]+=x;}
void pushdown(int p){modify(ls,tag[p]),modify(rs,tag[p]),tag[p]=0;}
void update(int p,int l,int r,int L,int R,int x)
{
if(R<l||r<L) return ;
if(L<=l&&r<=R) return modify(p,x);
if(tag[p]) pushdown(p);
update(ls,l,mid,L,R,x),update(rs,mid+1,r,L,R,x),pushup(p);
}
ll query(int p,int l,int r,int L)
{
if(r<L) return 0;
if(L<=l) return mx[p];
if(tag[p]) pushdown(p);
return max(query(ls,l,mid,L),query(rs,mid+1,r,L));
}
int main()
{
int n=read(),m=read();ll ans=0;
for(int i=1;i<=n;++i) a[i]=read();
for(int i=1;i<=m;++i) w[i]=read();
for(int i=n;i;--i) next[i]=t[a[i]]? t[a[i]]:n+1,t[a[i]]=i;
for(int l=n;l;--l)
{
update(1,1,n,l,next[l]-1,w[a[l]]);
if(next[next[l]]) update(1,1,n,next[l],next[next[l]]-1,-w[a[l]]);
ans=max(ans,query(1,1,n,l));
}
printf("%lld",ans);
}