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

线段树式暴力，从左向右扫描，遇到一个数，就把从这个位置到它下一次出现的位置减去它，下一个区间加上它。

维护区间加减，最值。

#include<algorithm>
#include<iostream>
#include<cstdio>
using namespace std;
const int N=1000005;
int n,m,f[N],pos[N],nxt[N];
long long ans,w[N];
struct node
{
	int l,r;
	long long mx,tag;
}a[4*N];
void build(int num,int l,int r)
{
	a[num].l=l,a[num].r=r;
	if(l==r)
		return ;
	int mid=(l+r)/2;
	build(2*num,l,mid);
	build(2*num+1,mid+1,r);
}
void update(int num)
{
	if(a[num].l!=a[num].r)
	{
		a[2*num].tag+=a[num].tag;
		a[2*num+1].tag+=a[num].tag;
		a[2*num].mx+=a[num].tag;
		a[2*num+1].mx+=a[num].tag;
	}
	a[num].tag=0;
}
void chg(int num,int l,int r,long long x)
{
	if(a[num].l>r||a[num].r<l)
		return ;
	if(a[num].l>=l&&a[num].r<=r)
	{
		a[num].mx+=x;
		a[num].tag+=x;
		return ;
	}
	if(a[num].tag!=0)
		update(num);
	chg(2*num,l,r,x);
	chg(2*num+1,l,r,x);
	a[num].mx=max(a[2*num].mx,a[2*num+1].mx);
}
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("%lld",&w[i]);
	for(int i=1;i<=n;i++)
		pos[i]=n+1;
	for(int i=n;i>=1;i--)
		nxt[i]=pos[f[i]],pos[f[i]]=i;
	build(1,1,n);
	for(int i=1;i<=m;i++)
		chg(1,pos[i],nxt[pos[i]]-1,w[i]);
	for(int i=1;i<=n;i++)
	{
		ans=max(ans,a[1].mx);
		chg(1,i,nxt[i]-1,-w[f[i]]);
		if(nxt[i]<=n)
			chg(1,nxt[i],nxt[nxt[i]]-1,w[f[i]]);
	}
	printf("%lld\n",ans);
	return 0;
}