BZOJ_3747_[POI2015]Kinoman_线段树

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。

---

 

预处理出来第$i$天下一次播放$f[i]$的日期$nxt[i]$，类似这道题http://www.cnblogs.com/suika/p/8890570.html

先求出左端点为$1$的情况，即出现第一次$+1$，第二次$-1$.

然后考虑删除掉$i$这个位置对答案的贡献，在$i+1\thicksim nxt[i]$的位置上减一，在$nxt[i]\thicksim nxt[nxt[i]]-1$的位置上加一

每次求出$i\thicksim n$的最大值，可以用线段树维护。

 

代码：

#include <stdio.h>
#include <string.h>
#include <algorithm>
using namespace std;
#define N 1000050
#define ls p<<1
#define rs p<<1|1
typedef long long ll;
char nc() {
	static char buf[100000],*p1,*p2;
	return p1==p2&&(p2=(p1=buf)+fread(buf,1,100000,stdin),p1==p2)?EOF:*p1++;
}
inline int rd() {
	register int x=0;
	register char s=nc();
	while(s<'0'||s>'9')s=nc();
	while(s>='0'&&s<='9')x=(x<<3)+(x<<1)+s-'0',s=nc();
	return x;
}
ll t[N<<2],add[N<<2];
int f[N],w[N],nxt[N],n,m,now[N];
void pushdown(int p) {
	ll d;
	if(d=add[p]) {
		add[ls]+=d; add[rs]+=d;
		t[ls]+=d; t[rs]+=d;
		add[p]=0;
	}
}
/*ll qmx(int l,int r,int x,int y,int p) {
	if(x<=l&&y>=r) return t[p];
	int mid=(l+r)>>11;
	ll re=0;
	pushdown(p);
	if(x<=mid) re=max(re,qmx(l,mid,x,y,ls));
	if(y>mid) re=max(re,qmx(mid+1,r,x,y,rs));
	return re; 
}*/
void update(int l,int r,int x,int y,ll v,int p) {
	if(x<=l&&y>=r) {
		t[p]+=v; add[p]+=v;
		return ;
	}
	pushdown(p);
	int mid=(l+r)>>1;
	if(x<=mid) update(l,mid,x,y,v,ls);
	if(y>mid) update(mid+1,r,x,y,v,rs);
	t[p]=max(t[ls],t[rs]);
}
int main() {
	n=rd(); m=rd();
	register int i;
	for(i=1;i<=n;i++) f[i]=rd();
	for(i=1;i<=m;i++) w[i]=rd();
	for(i=n;i;i--) nxt[i]=now[f[i]],now[f[i]]=i;
	for(i=1;i<=m;i++) {
		if(now[i]) {
			if(!nxt[now[i]]) update(1,n,now[i],n,w[i],1);
			else update(1,n,now[i],nxt[now[i]]-1,w[i],1);
		}
	}
	ll ans=0;
	for(i=1;i<=n;i++) {
		ans=max(ans,t[1]);
		if(nxt[i]) {
			update(1,n,i,nxt[i]-1,-w[f[i]],1);
			if(nxt[nxt[i]]) update(1,n,nxt[i],nxt[nxt[i]]-1,w[f[i]],1);
			else update(1,n,nxt[i],n,w[f[i]],1);
		}else update(1,n,i,n,-w[f[i]],1);
	}
	printf("%lld\n",ans);
}