[POI 2015]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

Hint

观看第2,3,4,5,6,7天内放映的电影，其中看且仅看过一次的电影的编号为2,3,4。

题解

这道题稀里糊涂的写了一个下午，思路什么的乱七八糟的...回家理清思路，半个小时就写完了。

我们枚举区间左端点，线段树维护每个位置作为右端点的答案，

$nex[i]$记录第$i$天的电影下次播放时间

当我们往右移的时候，显然$i$这个点的颜色已经不会为之后的区间有加成，我们需要将$i$~$nex[i]-1$这一段区间减掉$w[f[i]]$。

同样，当我们左端点右移的时候，将迎接一个新的$f[i]$，那么我们就要将$nex[i]$~$nex[nex[i]]-1$这一段区间加上$w[f[i]]$。

线段树维护，支持区间修改以及查询最大值。

注意一开始的时候就把所有颜色的最左的一段加入线段树中。

//It is made by Awson on 2017.10.16
#include <set>
#include <map>
#include <cmath>
#include <ctime>
#include <cmath>
#include <stack>
#include <queue>
#include <vector>
#include <string>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <iostream>
#include <algorithm>
#define LL long long
#define Min(a, b) ((a) < (b) ? (a) : (b))
#define Max(a, b) ((a) > (b) ? (a) : (b))
#define sqr(x) ((x)*(x))
#define Lr(o) (o<<1)
#define Rr(o) (o<<1|1)
using namespace std;
const int N = 1000000;
void read(int &x) {
    char ch; bool flag = 0;
    for (ch = getchar(); !isdigit(ch) && ((flag |= (ch == '-')) || 1); ch = getchar());
    for (x = 0; isdigit(ch); x = (x<<1)+(x<<3)+ch-48, ch = getchar());
    x *= 1-2*flag;
}

struct segment {
    LL sgm[(N<<2)+5], lazy[(N<<2)+5];
    void pushdown(int o) {
        sgm[Lr(o)] += lazy[o], sgm[Rr(o)] += lazy[o];
        lazy[Lr(o)] += lazy[o], lazy[Rr(o)] += lazy[o];
        lazy[o] = 0;
    }
    void update(int o, int l, int r, int a, int b, int key) {
        if (a <= l && r <= b) {
            sgm[o] += key, lazy[o] += key;
            return;
        }
        pushdown(o);
        int mid = (l+r)>>1;
        if (a <= mid) update(Lr(o), l, mid, a, b, key);
        if (b > mid) update(Rr(o), mid+1, r, a, b, key);
        sgm[o] = Max(sgm[Lr(o)], sgm[Rr(o)]);
    }
}T;

int n, m, f[N+5], w[N+5];
int path[N+5], nex[N+5];

void work() {
    read(n), read(m);
    for (int i = 1; i <= n; i++) read(f[i]);
    for (int i = 1; i <= m; i++) read(w[i]);
    for (int i = n; i >= 1; i--) {
        nex[i] = path[f[i]], path[f[i]] = i;
    }
    for (int i = 1; i <= m; i++) if (path[i]) {
        if (nex[path[i]]) T.update(1, 1, n, path[i], nex[path[i]]-1, w[i]);
        else T.update(1, 1, n, path[i], n, w[i]);
    }
    LL ans = 0;
    for (int i = 1; i <= n; i++) {
        ans = Max(ans, T.sgm[1]);
        if (nex[i]) {
            T.update(1, 1, n, i, nex[i]-1, -w[f[i]]);
            if (nex[nex[i]]) T.update(1, 1, n, nex[i], nex[nex[i]]-1, w[f[i]]);
            else T.update(1, 1, n, nex[i], n, w[f[i]]);
        }else T.update(1, 1, n, i, n, -w[f[i]]);
    }
    printf("%lld
", ans);
}
int main() {
    work();
    return 0;
}