BZOJ 3747: [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。

分析

设pre[i]表示与i这个位置上的数相等的前一个数的下标。
用线段树维护答案。
枚举右端点，每次加入一个数，那么pre[i] + 1到i这段位置的答案加上这个数的权值，pre[pre[i]] + 1到pre[i]这段位置的答案减去这个数的权值，然后查询[1, i]的最大值即可。

代码

#include <bits/stdc++.h>

#define N 1000005
#define ll long long

struct NOTE
{
    ll lazy;
    ll max;
}t[N << 2];

int n,m;
int kind[N], w[N];

int pre[N],next[N];

void pushUp(int p)
{
    t[p].max = std::max(t[p * 2].max, t[p * 2 + 1].max);
}

void pushDown(int p)
{
    if (t[p].lazy)
    {
        t[p * 2].lazy += t[p].lazy;
        t[p * 2 + 1].lazy += t[p].lazy;
        t[p * 2].max += t[p].lazy;
        t[p * 2 + 1].max += t[p].lazy;
        t[p].lazy = 0;
    }
}

void insert(int p,int l,int r,int x,int y,int c)
{
    if (x <= l && r <= y)
    {
        t[p].max += c;
        t[p].lazy += c;
        return;
    }
    int mid = (l + r) >> 1;
    pushDown(p);
    if (x <= mid)
        insert(p * 2, l, mid, x, y, c);
    if (y > mid)
        insert(p * 2 + 1, mid + 1, r, x, y, c);
    pushUp(p);
}

ll query(int p,int l,int r,int x,int y)
{
    if (x <= l && r <= y)
        return t[p].max;
    int mid = (l + r) >> 1;
    pushDown(p);
    ll res = 0;
    if (x <= mid)
        res = std::max(res, query(p * 2, l, mid, x, y));
    if (y > mid)
        res = std::max(res, query(p * 2 + 1, mid + 1, r, x, y));
    return res;
}

int main()
{
    scanf("%d%d",&n,&m);
    for (int i = 1; i <= n; i++)
        scanf("%d",&kind[i]);
    for (int i = 1; i <= m; i++)
        scanf("%d",&w[i]);
    for (int i = 1; i <= n; i++)
        pre[i] = next[kind[i]], next[kind[i]] = i;

    ll ans = 0;
    for (int i = 1; i <= n; i++)
    {
        insert(1, 1, n, pre[i] + 1, i, w[kind[i]]);
        if (pre[i])
            insert(1, 1, n, pre[pre[i]] + 1, pre[i], -w[kind[i]]);
        ans = std::max(ans, query(1, 1, n, 1, i));
    }

    printf("%lld\n",ans);
}