【bzoj 3747】[POI2015]Kinoman（线段树）

3747: [POI2015]Kinoman

Time Limit: 60 Sec  Memory Limit: 128 MB
Submit: 707  Solved: 291
[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

[Submit][Status][Discuss]

【题解】【线段树求区间最大值】

【记录区间时，用类似next数组的方式】

#include<cstdio>
#include<cstring>
#include<algorithm>
#define ll long long
using namespace std;
ll sum[5000010],delta[5000010],ans;
int n,m,a[1000010],nxt[1000010],last[1000010],p[1000010];
ll val[1000010];
inline void updata(int now)
{
	sum[now]=max(sum[now<<1],sum[now<<1|1]);
}
inline void pushdown(int now)
{
	if(delta[now])
	 {
	 	sum[now<<1]+=delta[now]; sum[now<<1|1]+=delta[now];
	 	delta[now<<1]+=delta[now]; delta[now<<1|1]+=delta[now];
	 	delta[now]=0;
	 }
}
void change(int now,int l,int r,int al,int ar,ll v)
{
	if(al<=l&&r<=ar)
	 {
	 	sum[now]+=v;
	 	delta[now]+=v;
	 	return;
	 }
	int mid=(l+r)>>1;
	pushdown(now);
	if(al<=mid) change((now<<1),l,mid,al,ar,v);
	if(ar>mid) change((now<<1)|1,mid+1,r,al,ar,v);
	updata(now);
}
int main()
{
	freopen("int.txt","r",stdin);
	freopen("my.txt","w",stdout);
	int i,j;
	scanf("%d%d",&n,&m);
	for(i=1;i<=n;++i)
	 {
	 	scanf("%d",&a[i]);
	 	nxt[p[a[i]]]=i;
	 	if(!p[a[i]]) last[a[i]]=i;
	 	p[a[i]]=i;
	 }
	for(i=1;i<=m;++i) scanf("%lld",&val[i]);
	for(i=1;i<=m;++i)
	 if(nxt[last[i]]&&last[i]) change(1,1,n,last[i],nxt[last[i]]-1,val[i]);
	  else
	   if(last[i]) change(1,1,n,last[i],n,val[i]);
	for(i=1;i<=n;++i)
	 {
	 	ans=max(ans,sum[1]);
	 	if(nxt[i]) change(1,1,n,i,nxt[i]-1,-val[a[i]]);
	 	 else change(1,1,n,i,n,-val[a[i]]);
	 	if(nxt[i])
	 	 if(nxt[nxt[i]]) change(1,1,n,nxt[i],nxt[nxt[i]]-1,val[a[i]]);
	 	  else change(1,1,n,nxt[i],n,val[a[i]]);
	 }
	printf("%lld\n",ans);
	return 0;
}