BZOJ3747 : [POI2015]Kinoman

首先预处理出[1,i]的答案

设nxt[i]表示i下一次出现的位置

随着左端点l往右移，[l+1,nxt[l]-1]里答案都减去它的价值，[nxt[l],nxt[nxt[l]]-1]里答案都加上它的价值

线段树维护

 

#include<cstdio>
#define N 1000010
typedef long long ll;
int n,m,i,f[N],w[N],nxt[N],loc[N],vis[N];ll ans[N],tag[N<<2],v[N<<2],fin;
inline void read(int&a){char c;while(!(((c=getchar())>='0')&&(c<='9')));a=c-'0';while(((c=getchar())>='0')&&(c<='9'))(a*=10)+=c-'0';}
inline ll max(ll a,ll b){return a>b?a:b;}
inline void up(int x){v[x]=max(v[x<<1],v[x<<1|1]);}
void build(int x,int a,int b){
  if(a==b){v[x]=ans[a];return;}
  int mid=(a+b)>>1;
  build(x<<1,a,mid),build(x<<1|1,mid+1,b);
  up(x);
}
inline void add1(int x,ll p){v[x]+=p,tag[x]+=p;}
void add(int x,int a,int b,int c,int d,int p){
  if(c<=a&&b<=d){add1(x,p);return;}
  if(tag[x])add1(x<<1,tag[x]),add1(x<<1|1,tag[x]),tag[x]=0;
  int mid=(a+b)>>1;
  if(c<=mid)add(x<<1,a,mid,c,d,p);
  if(d>mid)add(x<<1|1,mid+1,b,c,d,p);
  up(x);
}
int main(){
  read(n),read(m);
  for(i=1;i<=n;i++)read(f[i]);
  for(i=1;i<=m;i++)read(w[i]),loc[i]=n+1;
  for(i=n;i;i--)nxt[i]=loc[f[i]],loc[f[i]]=i;
  for(i=1;i<=n;vis[f[i++]]++)if(!vis[f[i]])ans[i]=ans[i-1]+w[f[i]];else if(vis[f[i]]==1)ans[i]=ans[i-1]-w[f[i]];else ans[i]=ans[i-1];
  build(1,1,n);
  fin=v[1];
  for(i=1;i<n;i++){
    add(1,1,n,i+1,nxt[i]-1,-w[f[i]]);
    if(nxt[i]<=n)add(1,1,n,nxt[i],nxt[nxt[i]]-1,w[f[i]]);
    if(fin<v[1])fin=v[1];
  }
  return printf("%lld",fin),0;
}