[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

输出观看且仅观看过一次的电影的好看值的总和的最大值。

这是一道好题，看了题解之后才会做。思路是这样的：数组next[i]表示：f[i]下次出现的位置,若无下次出现则为n+1。我们先处理出当1为左端点时的答案（也就是1-1、1-2、1-3……1-n的好看值总和）。接着我们考虑当左端点由i改为i+1的时候，右端点i到next[i-1]-1的答案就会减去w[f[i-1]]，右端点next[i-1]到next[next[i-1]]-1的答案会增加w[f[i-1]]。这样我们就可以用线段树维护，注意答案要开long long。时间复杂度为 O(nlogn)。

收获：以后看到有关区间出现次数这样的题目，一般要考虑开next数组，或者用莫队之类的东西。

代码：

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int maxn=1000010;
#define LL long long
int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch>='0'&&ch<='9'){x=(x<<3)+(x<<1)+ch-'0';ch=getchar();}
	return x*f;
}
int n,m,f[maxn],w[maxn],last[maxn],next[maxn],num[maxn],len=0;
//next[i]:f[i]下次出现的位置,若无下次出现则为n+1 
struct Tree{int l,r,lc,rc;LL c,lazy;}tr[maxn*2];
void build(int l,int r)
{
	int t=++len;
	tr[t].l=l;tr[t].r=r;tr[t].c=tr[t].lazy=0LL;
	if(l<r)
	{
		int mid=l+r>>1;
		tr[t].lc=len+1;build(l,mid);
		tr[t].rc=len+1;build(mid+1,r);
	}
}
void pushdown(int now)
{
	int lc=tr[now].lc,rc=tr[now].rc;
	tr[lc].lazy+=tr[now].lazy,tr[rc].lazy+=tr[now].lazy;
	tr[lc].c+=tr[now].lazy,tr[rc].c+=tr[now].lazy;
	tr[now].lazy=0LL;
}
void change(int now,int l,int r,LL x)
{
	if(tr[now].l==l&&tr[now].r==r)
	{
		tr[now].c+=x;tr[now].lazy+=x;
		return;
	}
	int mid=tr[now].l+tr[now].r>>1,lc=tr[now].lc,rc=tr[now].rc;
	if(tr[now].lazy!=0)pushdown(now);
	if(r<=mid)change(lc,l,r,x);
	else if(l>mid)change(rc,l,r,x);
	else change(lc,l,mid,x),change(rc,mid+1,r,x);
	tr[now].c=max(tr[lc].c,tr[rc].c);
}
int main()
{
	memset(num,0,sizeof(num));
	memset(last,-1,sizeof(last));
	memset(next,-1,sizeof(next));
	n=read();m=read();
	for(int i=1;i<=n;i++) f[i]=read();
	for(int i=1;i<=m;i++) w[i]=read();
	for(int i=1;i<=n;i++)
	{
		if(last[f[i]]!=-1)next[last[f[i]]]=i;
		last[f[i]]=i;
	}
	for(int i=1;i<=n;i++)if(next[i]==-1)next[i]=n+1;
	build(1,n);
	LL ans=0;
	for(int i=1;i<=n;i++)//处理左端点为1时的情况 
	{
		num[f[i]]++;
		if(num[f[i]]==1) ans+=(LL)(w[f[i]]);
		if(num[f[i]]==2) ans-=(LL)(w[f[i]]);
		change(1,i,i,ans);
	}
	for(int i=2;i<=n;i++)//枚举左端点变为i 
	{
		ans=max(ans,tr[1].c);
		if(i<=next[i-1]-1)change(1,i,next[i-1]-1,(LL)(-w[f[i-1]]));
		if(next[i-1]<=next[next[i-1]]-1)change(1,next[i-1],next[next[i-1]]-1,(LL)(w[f[i-1]]));
	}
	printf("%lld",ans);
}