bzoj3747【POI2015】Kinoman

3747: [POI2015]Kinoman

Time Limit: 60 Sec  Memory Limit: 128 MB
Submit: 474  Solved: 180
[Submit][Status][Discuss]

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

做法类似上一道题bzoj3339，线段树维护区间最大值和标记。

每次左端点从i右移到i+1时，将区间[i+1,next[i]-1]减去w[f[i]]，再将[next[i],next[next[i]]-1]加上w[f[i]]。

要注意next[i]和next[next[i]]不存在的情况。

#include<iostream>#include<algorithm>#include<cstdio>#include<cstdlib>#include<cmath>#include<cstring>#define F(i,j,n) for(int i=j;i<=n;i++)#define D(i,j,n) for(int i=j;i>=n;i--)#define LL long long#define pa pair<int,int>#define MAXN 1000005using namespace std;struct tree_type{int l,r;LL w,mx;}t[MAXN*4];int n,m,a[MAXN],b[MAXN],cnt[MAXN],nxt[MAXN],last[MAXN];LL ans=0,f[MAXN];int read(){int ret=0,flag=1;char ch=getchar();while (ch<'0'||ch>'9'){if (ch=='-') flag=-1;ch=getchar();}while (ch>='0'&&ch<='9'){ret=ret*10+ch-'0';ch=getchar();}return ret*flag;}void build(int k,int x,int y){t[k].l=x;t[k].r=y;if (x==y){t[k].w=f[x];t[k].mx=f[x];return;}t[k].w=0;int mid=(x+y)>>1;build(k<<1,x,mid);build(k<<1|1,mid+1,y);t[k].mx=max(t[k<<1].mx,t[k<<1|1].mx);}void pushdown(int k){t[k<<1].w+=t[k].w;t[k<<1].mx+=t[k].w;t[k<<1|1].w+=t[k].w;t[k<<1|1].mx+=t[k].w;t[k].w=0;}void update(int k,int x){t[k].w+=x;t[k].mx+=x;}void add(int k,int x,int y,int z){if (t[k].l==x&&t[k].r==y){update(k,z);return;}pushdown(k);int mid=(t[k].l+t[k].r)>>1;if (mid>=y) add(k<<1,x,y,z);else if (mid+1<=x) add(k<<1|1,x,y,z);else add(k<<1,x,mid,z),add(k<<1|1,mid+1,y,z);t[k].mx=max(t[k<<1].mx,t[k<<1|1].mx);}int main(){memset(cnt,0,sizeof(cnt));memset(last,0,sizeof(last));n=read();m=read();F(i,1,n) a[i]=read();F(i,1,m) b[i]=read();F(i,1,n){if (cnt[a[i]]==0) f[i]=f[i-1]+b[a[i]];else if (cnt[a[i]]==1) f[i]=f[i-1]-b[a[i]];else f[i]=f[i-1];cnt[a[i]]++;}D(i,n,1) nxt[i]=last[a[i]],last[a[i]]=i;build(1,1,n);F(i,1,n){ans=max(ans,t[1].mx);int x=nxt[i];if (x){add(1,i,x-1,-b[a[i]]);if (nxt[x]) add(1,x,nxt[x]-1,b[a[i]]);else add(1,x,n,b[a[i]]);}else add(1,i,n,-b[a[i]]);}printf("%lld\n",ans);}