【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 ;

}

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