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
(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
,w
,nxt
,n,m,now
;
void pushdown(int p) {
ll d;
if(d=add
) {
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);
}

[p]