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);
}