bzoj3747[POI2015]Kinoman

3747: [POI2015]Kinoman

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

可以固定左端点,单点查询后面的区间最大值
记录链表每部电影出现的位置,首先对于1到每部电影第一次出现的位置加上价值
然后考虑i->i+1   i到next[i]-1区间整体减小w[f[i]] next[i]到next[next[i]]-1整体加上w[f[i]]

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
#define ls u<<1
#define rs u<<1|1
#define ll long long
#define N 1000050
using namespace std;
ll mx[N<<2],lz[N<<2];int n,m,hd[N],next[N],w[N],f[N];
void pushdown(int u){
    if(!lz[u])return;
    ll x=lz[u];
    lz[ls]+=x;lz[rs]+=x;
    mx[ls]+=x;mx[rs]+=x;
    lz[u]=0;
}
void pushup(int u){mx[u]=max(mx[ls],mx[rs]);}
void update(int u,int L,int R,int l,int r,int val){
    if(l<=L&&R<=r){
        lz[u]+=val;
        mx[u]+=val;
        return;
    }
    pushdown(u);int mid=L+R>>1;
    if(l<=mid)update(ls,L,mid,l,r,val);
    if(r>mid)update(rs,mid+1,R,l,r,val);
    pushup(u);
}
int main(){
    scanf("%d%d",&n,&m);
    for(register int i=1;i<=n;i++)scanf("%d",&f[i]);
    for(int i=1;i<=m;i++)scanf("%d",&w[i]);
    for(register int i=n;i;i--){
        next[i]=hd[f[i]];
        hd[f[i]]=i;
    }
    for(int i=1;i<=m;i++){
        if(!hd[i])continue;
        if(!next[hd[i]])update(1,1,n,hd[i],n,w[i]);
        else update(1,1,n,hd[i],next[hd[i]]-1,w[i]);
    }
    ll ans=0;
    for(register int p=1;p<=n;p++){
        ans=max(ans,mx[1]);
        int t=next[p];
        if(!t)update(1,1,n,p,n,-w[f[p]]);
        else{
            update(1,1,n,p,t-1,-w[f[p]]);
            if(next[t])update(1,1,n,t,next[t]-1,w[f[p]]);
            else update(1,1,n,t,n,w[f[p]]);
        }
    }
    cout<<ans;
    return 0;
}