【BZOJ-3747】Kinoman 线段树

3747: [POI2015]Kinoman

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

Solution

经典题，然而想了一会.

我们先预处理出$suf[i]$表示$i$这个位置放的电影，和它同样的电影下一次放的位置。

然后我们枚举左端点，不断更新答案就行。

Code

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<cmath>
using namespace std;
#define LL long long
inline int read()
{
    int x=,f=; char ch=getchar();
    while (ch<'' || ch>'') {if (ch=='-') f=-; ch=getchar();}
    while (ch>='' && ch<='') {x=x*+ch-''; ch=getchar();}
    return x*f;
}
#define MAXN 1000100
int N,M;
namespace SegmentTree
{
    struct SegmentTreeNode{int l,r; LL tag,maxx;}tree[MAXN<<];
    #define ls now<<1
    #define rs now<<1|1
    inline void Update(int now) {tree[now].maxx=max(tree[ls].maxx,tree[rs].maxx);}
    inline void PushDown(int now)
    {
        if (tree[now].l==tree[now].r || !tree[now].tag) return;
        LL D=tree[now].tag; tree[now].tag=;
        tree[ls].maxx+=D; tree[ls].tag+=D;
        tree[rs].maxx+=D; tree[rs].tag+=D;
    }
    inline void BuildTree(int now,int l,int r)
    {
        tree[now].l=l; tree[now].r=r;
        if (l==r) return;
        int mid=(l+r)>>;
        BuildTree(ls,l,mid); BuildTree(rs,mid+,r);
        Update(now);
    }
    inline void Modify(int now,int L,int R,int D)
    {
        if (L>R) return;
        int l=tree[now].l,r=tree[now].r;
        PushDown(now);
        if (L<=l && R>=r) {tree[now].maxx+=D; tree[now].tag+=D; return;}
        int mid=(l+r)>>;
        if (L<=mid) Modify(ls,L,R,D);
        if (R>mid) Modify(rs,L,R,D);
        Update(now);
    }
}dddddd
int suf[MAXN],last[MAXN],f[MAXN],w[MAXN],first[MAXN];
LL ans;
int main()
{
    N=read(),M=read();
    for (int i=; i<=N; i++) f[i]=read();
    for (int i=; i<=M; i++) w[i]=read();
    for (int i=; i<=N; i++)
        suf[last[f[i]]]=i,last[f[i]]=i;
    for (int i=; i<=N; i++)
        if (!first[f[i]]) first[f[i]]=i;
    SegmentTree::BuildTree(,,N);
    for (int i=; i<=M; i++)
        if (first[i])
            SegmentTree::Modify(,first[i],suf[first[i]]? suf[first[i]]-:N,w[i]);
    for (int i=; i<=N; i++)
        {
            ans=max(ans,SegmentTree::tree[].maxx);
            SegmentTree::Modify(,i,suf[i]? suf[i]-:N,-w[f[i]]);
            if (suf[i]) SegmentTree::Modify(,suf[i],suf[suf[i]]? suf[suf[i]]-:N,w[f[i]]); else continue;
        }
    printf("%lld\n",ans);
    return ;
}

经典题我居然现在才做....