bzoj 3747: [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。

HINT

Source

鸣谢Jcvb

 

这道题是一道很经典的线段树题目了；

题目没有给定询问区间，如果给定询问区间的话那就是用类似采花的做法就行了；

我们考虑从后往前枚举左端点，那么我们相当于只需要查询右端点的最值；

我们考虑记nxt[i]，表示下一个跟i同颜色的位置，那么从后往前枚举左端点的话，

右端点在[i,nxt[i]-1]间的贡献会+w,然后我们需要求的是仅出现一次的，

那么原来在[nxt[i],nxt[nxt[i]]-1]的右端点的贡献需要-w，至于[nxt[nxt[i]],nxt[nxt[nxt[i]]]]的右端点必定在之前处理过了不用管；

所以我们相当于区间修改和查询最大值即可，用线段树来实现；

//MADE BY QT666
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#define lson x<<1
#define rson x<<1|1
using namespace std;
typedef long long ll;
const int N=1500050;
int n,m,w[N],f[N],nxt[N],last[N];
ll tr[N*4],lazy[N*4];
void update(int x,int l,int r,int xl,int xr,int v){
    if(xl>xr) return;
    if(xl<=l&&r<=xr){
	tr[x]+=v;lazy[x]+=v;return;
    }
    int mid=(l+r)>>1;
    if(xr<=mid) update(lson,l,mid,xl,xr,v);
    else if(xl>mid) update(rson,mid+1,r,xl,xr,v);
    else update(lson,l,mid,xl,mid,v),update(rson,mid+1,r,mid+1,xr,v);
    tr[x]=max(tr[lson],tr[rson])+lazy[x];
}
ll query(int x,int l,int r,int xl,int xr,int la){
    if(xl<=l&&r<=xr) return tr[x]+la;
    int mid=(l+r)>>1;la+=lazy[x];
    if(xr<=mid) return query(lson,l,mid,xl,xr,la);
    else if(xl>mid) return query(rson,mid+1,r,xl,xr,la);
    else return max(query(lson,l,mid,xl,mid,la),query(rson,mid+1,r,mid+1,xr,la));
}
int main(){
    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;i--) nxt[i]=last[f[i]],last[f[i]]=i;
    ll ans=0;
    for(int i=n;i;i--){
	if(!nxt[i]) update(1,1,n,i,n,w[f[i]]);
	else {
	    update(1,1,n,i,nxt[i]-1,w[f[i]]);
	    if(nxt[nxt[i]]) update(1,1,n,nxt[i],nxt[nxt[i]]-1,-w[f[i]]);
	    else update(1,1,n,nxt[i],n,-w[f[i]]);
	}
	ans=max(ans,query(1,1,n,i,n,0));
    }
    printf("%lld\n",ans);
    return 0;
}