Bzoj3747 [POI1015] 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。

Bzoj权限题。只过了样例，尚未评测。

题面搬运自神犇hzwer博客。

线段树维护区间最值，枚举区间起点，不断把起点前面的电影贡献值删去，再把当前电影“从当前直到它下次出现之前”的区间都新加上当前电影的贡献值，不断更新最大答案。

刚开始按区间加法写的，所以变量用了sum，写到一半发现只搞区间最大值就好，然而懒得改变量名了……

 #include<algorithm>
 #include<iostream>
 #include<cstring>
 #include<cstdio>
 #include<cmath>
 #define ls l,mid,rt<<1
 #define rs mid+1,r,rt<<1|1
 #define LL long long
 using namespace std;
 const int mxn=;
 int n,m;
 int f[mxn],w[mxn];
 int next[mxn],last[mxn];
 int cnt[mxn];
 struct node{
     LL sum;//变量名是sum，实际是存区间最大值
     LL tag;
 }t[mxn<<];
 void Build(int l,int r,int rt){//init //实际并没有用到
     if(l==r){
         t[rt].sum=t[rt].tag=;
         return;
     }
     int mid=(l+r)>>;
     Build(ls);Build(rs);
     return;
 }
 void pushdown(int l,int r,int rt){
     if(l==r)return;
     if(t[rt].tag){
         int mid=(l+r)>>;
         t[rt<<].tag+=t[rt].tag;
         t[rt<<|].tag+=t[rt].tag;
         t[rt<<].sum+=t[rt].tag;
         t[rt<<|].sum+=t[rt].tag;
         t[rt].tag=;
     }
 }
 void add(int L,int R,int v,int l,int r,int rt){
     if(t[rt].tag)pushdown(l,r,rt);
     if(L<=l && r<=R){
         t[rt].tag+=v;
         t[rt].sum+=v;
         return;
     }
     int mid=(l+r)>>;
     if(L<=mid)add(L,R,v,ls);
     if(R>mid)add(L,R,v,rs);
     t[rt].sum=max(t[rt<<].sum,t[rt<<|].sum);
     return;
 }
 void read(){
     scanf("%d%d",&n,&m);
     int i,j;
     for(i=;i<=n;i++)scanf("%d",&f[i]);
     for(i=;i<=m;i++)scanf("%d",&w[i]);
     return;
 }
 int main(){
     read();
     int i,j;
     for(i=n;i;i--){
         next[i]=last[f[i]];
         last[f[i]]=i;
     }
     for(i=;i<=m;i++){
         if(last[i]){
             if(!next[last[i]])add(last[i],n,w[i],,n,);
             else add(last[i],next[last[i]]-,w[i],,n,);
         }
     }
     LL ans=;
     for(i=;i<=n;i++){
         ans=max(ans,t[].sum);
         int tmp=next[i];
         if(tmp){
             add(i,tmp-,-w[f[i]],,n,);
             if(next[tmp])add(tmp,next[tmp]-,w[f[i]],,n,);
             else add(tmp,n,w[f[i]],,n,);//如果之后没有相同电影了，从next[i]到n的区间都加值
         }
         else add(i,n,-w[f[i]],,n,);//删除上一个区间的值
     }
     printf("%lld",ans);
     return ;
 }