Bzoj 3747: [POI2015]Kinoman 线段树

3747: [POI2015]Kinoman

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

题解：

线段树。。。

用last[]记录每个位置的电影的上一个位置。

然后从头到尾依次遍历，每次把 (当前位置的电影的上一个位置,当前位置] 这个区间加上当前电影的好看值。然后把 (当前位置的电影的上上一个位置,当前位置的电影的上一个位置] 这个区间减去当前电影的好看值。每次求区间最大值即可。

这个用线段树维护即可。。。

 #include<bits/stdc++.h>
 using namespace std;
 #define INF 1e9
 #define LL long long
 struct node
 {
     int left,right;
     LL tag,mx;
 }tree[];
 int f[],w[],last[],pre[];
 int read()
 {
     int s=,fh=;char ch=getchar();
     while(ch<''||ch>''){if(ch=='-')fh=-;ch=getchar();}
     while(ch>=''&&ch<=''){s=s*+(ch-'');ch=getchar();}
     return s*fh;
 }
 void Build(int k,int l,int r)
 {
     tree[k].left=l;tree[k].right=r;tree[k].tag=;tree[k].mx=;
     if(l==r)return;
     int mid=(l+r)/;
     Build(k*,l,mid);Build(k*+,mid+,r);
 }
 void Pushup(int k)
 {
     tree[k].mx=max(tree[k*].mx,tree[k*+].mx);
 }
 void Pushdown(int k)
 {
     if(tree[k].tag!=)
     {
         tree[k*].tag+=tree[k].tag;
         tree[k*+].tag+=tree[k].tag;
         tree[k*].mx+=tree[k].tag;
         tree[k*+].mx+=tree[k].tag;
         tree[k].tag=;
     }
 }
 void Add(int k,int l,int r,int add)
 {
     if(l<=tree[k].left&&tree[k].right<=r)
     {
         tree[k].tag+=(LL)add;
         tree[k].mx+=(LL)add;
         return;
     }
     Pushdown(k);
     int mid=(tree[k].left+tree[k].right)/;
     if(r<=mid)Add(k*,l,r,add);
     else if(l>mid)Add(k*+,l,r,add);
     else {Add(k*,l,mid,add);Add(k*+,mid+,r,add);}
     Pushup(k);
 }
 int main()
 {
     int n,m,i;
     LL ans;
     n=read();m=read();
     for(i=;i<=n;i++)f[i]=read();
     for(i=;i<=m;i++)w[i]=read();
     memset(last,,sizeof(last));
     memset(pre,,sizeof(pre));
     for(i=;i<=n;i++)
     {
         last[i]=pre[f[i]];
         pre[f[i]]=i;
     }
     Build(,,n);
     ans=-INF;
     for(i=;i<=n;i++)
     {
         Add(,last[i]+,i,w[f[i]]);
         if(last[i]!=)
         {
             Add(,last[last[i]]+,last[i],-w[f[i]]);
         }
         ans=max(ans,tree[].mx);
     }
     printf("%lld",ans);
     return ;
 }