【bzoj3747】[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],…,fn。
第三行包含m个整数w[1],w[2],…,wm。
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。
题解
线段树经典题
nxt[i]记录第i天的电影下次播放时间
枚举区间左端点，线段树维护每个位置作为右端点的答案
考虑l-r的左端点变为l+1
发现l到nxt[l]-1的答案减少w[f[l]]
而nxt[l]到nxt[nxt[l]]-1增加w[f[l]]
by hzwer

代码

#include<bits/stdc++.h>
#define inf 2147483647
#define N 400005
#define M 1000005
#define ll long long
using namespace std;
inline int read()
{
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
int n,m;
int f[1000005],w[1000005];
int last[1000005],nxt[1000005];
ll ans;
struct node{int l,r;ll tag,mx;}t[4000005];
inline void pushdown(int k,int l,int r)
{
    if (l==r) return;
    t[k<<1].tag+=t[k].tag;t[k<<1|1].tag+=t[k].tag;
    t[k<<1].mx+=t[k].tag;t[k<<1|1].mx+=t[k].tag;
    t[k].tag=0;
}
void modify(int k,int l,int r,int x,int y,int z)
{
    if (t[k].tag) pushdown(k,l,r);
    if (l==x&&r==y)
    {
        t[k].tag+=z;t[k].mx+=z;return;
    }
    int mid=(l+r)>>1;
    if (y<=mid) modify(k<<1,l,mid,x,y,z);
    else if (x>mid) modify(k<<1|1,mid+1,r,x,y,z);
    else {modify(k<<1,l,mid,x,mid,z);modify(k<<1|1,mid+1,r,mid+1,y,z);}
    t[k].mx=max(t[k<<1].mx,t[k<<1|1].mx);
}

int main()
{
    n=read();m=read();
    for (int i=1;i<=m;i++) last[i]=n+1;
    for (int i=1;i<=n;i++) f[i]=read();
    for (int i=n;i;i--)
    {
        nxt[i]=last[f[i]];
        last[f[i]]=i;
    }
    for (int i=1;i<=m;i++)
    {
        w[i]=read();
        if (last[i]<=n) modify(1,1,n,last[i],nxt[last[i]]-1,w[i]);
    }
    for (int i=1;i<=n;i++)
    {
        ans=max(ans,t[1].mx);
        modify(1,1,n,i,nxt[i]-1,-w[f[i]]);
        if (nxt[i]<=n) modify(1,1,n,nxt[i],nxt[nxt[i]]-1,w[f[i]]);
    }
    printf("%lld",ans);
    return 0;
}