BZOJ 3747 洛谷 3582 [POI2015]Kinoman

【题解】

　　扫描线+线段树。

　　我们记第i部电影上次出现的位置是$pre[i]$，我们从$1$到$n$扫描，每次区间$(pre[i],i]$加上第i部电影的贡献$w[f[i]]$，区间$[pre[pre[i]],pre[i]]$减去第i部电影的贡献$w[f[i]]$.

　　

 #include<cstdio>
 #include<algorithm>
 #define N 1000010
 #define rg register
 #define ls (u<<1)
 #define rs (u<<1|1)
 #define mid ((l+r)>>1)
 #define LL long long
 using namespace std;
 int n,m,f[N],w[N],pos[N],pre[N];
 LL ans;
 struct tree{
     LL mx,del;
 }a[N<<];
 inline int read(){
     int k=,f=; char c=getchar();
     while(c<''||c>'')c=='-'&&(f=-),c=getchar();
     while(''<=c&&c<='')k=k*+c-'',c=getchar();
     return k*f;
 }
 inline LL max(LL x,LL y){
     return x>y?x:y;
 }
 inline void pushup(int u){
     a[u].mx=max(a[ls].mx,a[rs].mx);
 }
 inline void pushdown(int u){
     if(!a[u].del) return;
     LL d=a[u].del; a[u].del=;
     a[ls].mx+=d; a[ls].del+=d;
     a[rs].mx+=d; a[rs].del+=d;
 }
 void add(int u,int ql,int qr,int l,int r,LL d){
     //printf("%d %d %d %d\n",ql,qr,l,r);
     if(ql<=l&&r<=qr){
         a[u].del+=d; a[u].mx+=d; return;
     }
     pushdown(u);
     if(ql<=mid) add(ls,ql,qr,l,mid,d);
     if(qr>mid) add(rs,ql,qr,mid+,r,d);
     pushup(u);
 }
 int main(){
     n=read(); m=read();
     for(int i=;i<=n;i++){
         f[i]=read();
         pre[i]=pos[f[i]]; pos[f[i]]=i;
     }
     for(int i=;i<=m;i++) w[i]=read();
     for(int i=;i<=n;i++){
         int prefix=pre[i];
         add(,prefix+,i,,n,w[f[i]]);
         if(prefix) add(,pre[prefix]+,prefix,,n,-w[f[i]]);
         ans=max(ans,a[].mx);
     }
     printf("%lld\n",ans);
     return ;
 }