bzoj 3747: [POI2015]Kinoman（线段树）

3747: [POI2015]Kinoman

Time Limit: 60 Sec  Memory Limit: 128 MB
Submit: 706  Solved: 290
[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]
HOME Back

题解：线段树。

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#define N 1000003#define LL long long using namespace std; LL delta[N*4],tr[N*4],n,m;LL v[N],col[N];int next[N],point[N],pos[N];void update(int now){tr[now]=max(tr[now<<1],tr[now<<1|1]);}void pushdown(int now,int l,int r){int mid=(l+r)/2;if (delta[now]) { tr[now<<1]+=delta[now]; tr[now<<1|1]+=delta[now]; delta[now<<1]+=delta[now]; delta[now<<1|1]+=delta[now]; delta[now]=0; }}void qjchange(int now,int l,int r,int ll,int rr,LL v){if (ll<=l&&r<=rr){delta[now]+=v;tr[now]+=v;return;}int mid=(l+r)/2;pushdown(now,l,r);if (ll<=mid) qjchange(now<<1,l,mid,ll,rr,v);if (rr>mid)  qjchange(now<<1|1,mid+1,r,ll,rr,v);update(now);}int main(){freopen("a.in","r",stdin);freopen("my.out","w",stdout);scanf("%lld%lld",&n,&m);for (int i=1;i<=n;i++) { scanf("%lld",&v[i]);  next[point[v[i]]]=i;  if (!point[v[i]])  pos[v[i]]=i;point[v[i]]=i; }for (int i=1;i<=m;i++) scanf("%lld",&col[i]);for (int i=1;i<=m;i++) if (next[pos[i]]&&pos[i]) qjchange(1,1,n,pos[i],next[pos[i]]-1,col[i]); else if (pos[i]) qjchange(1,1,n,pos[i],n,col[i]);LL ans=0;for (int i=1;i<=n;i++) { ans=max(ans,tr[1]);  if (next[i]) qjchange(1,1,n,i,next[i]-1,-col[v[i]]); else qjchange(1,1,n,i,n,-col[v[i]]); if (next[i]) { if (next[next[i]])  qjchange(1,1,n,next[i],next[next[i]]-1,col[v[i]]); else  qjchange(1,1,n,next[i],n,col[v[i]]);    } }printf("%lld\n",ans);}