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 1000005
using 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);
}