bzoj3747 [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],…,f[n](1<=f[i]<=m)。

第三行包含m个整数w[1],w[2],…,w[m](1<=w[j]<=1000000)。

Output

输出观看且仅观看过一次的电影的好看值的总和的最大值。

Sample Input

9 4
2 3 1 1 4 1 2 4 1
5 3 6 6

Sample Output

15
样例解释：
观看第2,3,4,5,6,7天内放映的电影，其中看且仅看过一次的电影的编号为2,3,4。

正解：套路＋线段树。

经典套路不用多说。。

考虑一下每个点的贡献，记一个电影的上一个播放天为$lst$，下一个播放天为$nxt$。

那么可以发现，一个点只有在$lst<l$且$r<nxt$的区间才有贡献。

于是把所有点按照$lst$排序，枚举左端点$l$，维护一个以右端点为下标的线段树，然后每次把$lst<l$的点加入贡献，并把$i<l$的点删除，这些就是区间修改，每次线段树维护全局最大值即可。

#include <bits/stdc++.h>
#define il inline
#define RG register
#define ll long long
#define N (1000010)
#define ls (x<<1)
#define rs (x<<1|1)

using namespace std;

int lst[N],nxt[N],q[N],f[N],w[N],cnt[N],n,m;
ll sum[N<<2],tag[N<<2],ans;

il int gi(){
  RG int x=0,q=1; RG char ch=getchar();
  while ((ch<'0' || ch>'9') && ch!='-') ch=getchar();
  if (ch=='-') q=-1,ch=getchar();
  while (ch>='0' && ch<='9') x=x*10+ch-48,ch=getchar();
  return q*x;
}

il int cmp(const int &a,const int &b){ return lst[a]<lst[b]; }

il void pushdown(RG int x){
  sum[ls]+=tag[x],sum[rs]+=tag[x];
  tag[ls]+=tag[x],tag[rs]+=tag[x];
  tag[x]=0; return;
}

il void update(RG int x,RG int l,RG int r,RG int xl,RG int xr,RG int v){
  if (xl<=l && r<=xr){ sum[x]+=v,tag[x]+=v; return; }
  if (tag[x]) pushdown(x); RG int mid=(l+r)>>1;
  if (xr<=mid) update(ls,l,mid,xl,xr,v);
  else if (xl>mid) update(rs,mid+1,r,xl,xr,v);
  else update(ls,l,mid,xl,mid,v),update(rs,mid+1,r,mid+1,xr,v);
  sum[x]=max(sum[ls],sum[rs]); return;
}

int main(){
#ifndef ONLINE_JUDGE
  freopen("kinoman.in","r",stdin);
  freopen("kinoman.out","w",stdout);
#endif
  n=gi(),m=gi();
  for (RG int i=1;i<=n;++i)
    q[i]=i,f[i]=gi(),nxt[cnt[f[i]]]=i,lst[i]=cnt[f[i]],nxt[i]=n+1,cnt[f[i]]=i;
  for (RG int i=1;i<=m;++i) w[i]=gi(); sort(q+1,q+n+1,cmp);
  for (RG int l=1,pos=1;l<=n;++l){
    for (RG int x=q[pos];pos<=n && lst[x]<l && x>=l;++pos,x=q[pos])
      update(1,1,n,x,nxt[x]-1,w[f[x]]);
    if (l>1) update(1,1,n,l-1,nxt[l-1]-1,-w[f[l-1]]); ans=max(ans,sum[1]);
  }
  cout<<ans; return 0;
}