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。

Solution

经典的线段树题目。

next[i]表示和i点下一场相同电影的位置。

枚举左端点,对于每一次左端点向右移动1,i…next[i]-1的所有天的答案都要小w[f[i]](因为第i天不看了嘛),next[i]…next[next[i]]-1的所有答案都要大w[f[i]](显然的,我们可以在这些点看f[i]这种电影)。

这样的话维护一下这个max数组就好了,然后每次枚举左端点更新ans。

维护的话用线段树维护就好了。 VerySlow…

时间有点慢啊,哪位大犇告诉蒟蒻怎么写一个常数小点的线段树啊…

Code

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
#include <iostream>
#include <cstdio>
using namespace std;
const int INF = 1<<30;
const int maxn = 1000005;
int getint() {
	int r = 0, k = 1; char c = getchar();
	for (; '0' > c || c > '9'; c = getchar()) if (c == '-') k = -1;
	for (; '0' <= c && c <= '9'; c = getchar()) r = r * 10 - '0' + c;
	return r * k;
}
typedef long long LL;
int n, m, f[maxn], w[maxn], next[maxn], h[maxn];
LL tag[maxn*5], mx[maxn*5], ans = 0;
void pu(int now) { mx[now] = max(mx[now<<1], mx[now<<1|1]); }
void pd(int now) {
	mx[now<<1] += tag[now];
	mx[now<<1|1] += tag[now];
	tag[now<<1] += tag[now];
	tag[now<<1|1] += tag[now];
	tag[now] = 0;
}
void change(int now, int l, int r, int ll, int rr, int x) {
	if (ll <= l && r <= rr) {
		if (l != r) tag[now] += x;
		mx[now] += x;
		return;
	}
	if (tag[now]) pd(now);
	int mid = (l+r)>>1;
	if (ll <= mid) change(now<<1, l, mid, ll, rr, x);
	if (rr > mid) change(now<<1|1, mid+1, r, ll, rr, x);
	pu(now);
}
int main() {
	n = getint(); m = getint();
	for (int i = 1; i <= n; ++i) f[i] = getint();
	for (int i = 1; i <= m; ++i) w[i] = getint();
	for (int i = n; i >= 1; --i) {
		next[i] = h[f[i]];
		h[f[i]] = i;
	}
	for (int i = 1; i <= m; ++i)
		if (h[i]) {
			if (next[h[i]]) change(1, 1, n, h[i], next[h[i]]-1, w[i]);
			else change(1, 1, n, h[i], n, w[i]);
		}
	int tmp;
	for (int i = 1; i <= n; ++i) {
		ans = max(ans, mx[1]);
		tmp = next[i];
		if (tmp) {
			change(1, 1, n, i, tmp-1, -w[f[i]]);
			if (next[tmp]) change(1, 1, n, tmp, next[tmp]-1, w[f[i]]);
			else change(1, 1, n, tmp, n, w[f[i]]);
		}
		else change(1, 1, n, i, n, -w[f[i]]);
	}
	printf("%lld", ans);
	return 0;
}