[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 42 3 1 1 4 1 2 4 15 3 6 6

Sample Output

15

样例解释：

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

题解：
如果我们事先预处理出所有 (1,i) 区间的答案并存在a[i]中的话，对于左端点在1的情况，我们只需要一个线段树查询所有的a[i]的最大值就好了。那么对于每一个左端点我们如何查询答案呢？
假设我们知道了以i为左端点的答案所构建的线段树的话，那么我们知道以i+1为左端点的答案线段树只需要在i的基础上将区间(i,nt[i])（nt[i]表示下一个和第i个电影相同的电影的位置）的答案减去w[f[i]]，然后将区间[nt[i],nt[nt[i]])的答案+w[f[i]]就好了（当然如果当前为这个电影最后一次放映的话只要直接统计到n就OK了）
当然对于左端点为i的情况来说，可能答案区间[1,i)的答案也会被统计入最小值，不过那些答案都无所谓，它们都是左端点<i的区间的某个答案(显然可知)
最后祝您，注意long long

#include<iostream>#include<cstring>#include<cstdlib>#include<cstdio>#include<cmath>#include<algorithm>#define ll long long#define LiangJiaJun mainusing namespace std;int f[1000004],n,m,w[1000004];struct segment{    int l,r;ll w,tag;}tr[4000004];int nt[1000004],bf[1000004];ll pre[1000004];int vis[1000004];void pushdown(int k){     ll x=tr[k].tag;     tr[k].tag=0;     tr[k<<1].w+=x;tr[k<<1].tag+=x;     tr[k<<1|1].w+=x;tr[k<<1|1].tag+=x;}void build(int k,int l,int r){     tr[k].l=l;tr[k].r=r;     if(l==r)return;     int mid=(l+r)>>1;     build(k<<1,l,mid);build(k<<1|1,mid+1,r);     return;}void modify(int k,int a,int b,ll x){     if(b<a)return;     if(tr[k].tag&&tr[k].l!=tr[k].r)pushdown(k);     int l=tr[k].l,r=tr[k].r;     if(l==a&&r==b){        tr[k].w+=x;        tr[k].tag+=x;        return;     }     int mid=(l+r)>>1;     if(b<=mid)modify(k<<1,a,b,x);     else if(a>mid)modify(k<<1|1,a,b,x);     else{        modify(k<<1,a,mid,x);        modify(k<<1|1,mid+1,b,x);     }     tr[k].w=max(tr[k<<1].w,tr[k<<1|1].w);     return;}int LiangJiaJun(){    memset(vis,0,sizeof(vis));    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--){        nt[i]=bf[f[i]];        bf[f[i]]=i;    }    build(1,1,n);    for(int i=1;i<=m;i++)if(bf[i])modify(1,bf[i],nt[bf[i]]-1>0?nt[bf[i]]-1:n,w[i]);    ll ans=0;    for(int i=1;i<=n;i++){        ans=max(ans,tr[1].w);        if(nt[i]){            modify(1,i+1,nt[i]-1,-w[f[i]]);            modify(1,nt[i],nt[nt[i]]-1>0?nt[nt[i]]-1:n,w[f[i]]);        }else{            modify(1,i+1,n,-w[f[i]]);        }    }    printf("%lld\n",ans);    return 0;}