【LG3582】[POI2015]KIN

【LG3582】[POI2015]KIN

题面

洛谷

题解

这题维护区间的信息有点像最大子段和，我们往最大子段和上面靠。

对于一个颜色，我们有一个直观的想法就是将它一次出现的权值设为正，二次出现就设为负。

但是当区间中出现次数多于两次怎么办呢？我们可以考虑从左往右扫这个序列，这个数出现在最靠右的位置时设权值为正，第二靠右出现时设权值为负，否则全部设为0，这样子的话，我们用最大子段和维护就行了。就算你只访问到一个负的权值也没关系，因为你在前面已经把这种情况的答案算进去了。

代码

#include <iostream>
#include <cstdio>
#include <cstdlib>
#include <cstring> 
#include <cmath> 
#include <algorithm>
using namespace std; 
inline int gi() {
    register int data = 0, w = 1;
    register char ch = 0;
    while (!isdigit(ch) && ch != '-') ch = getchar(); 
    if (ch == '-') w = -1, ch = getchar(); 
    while (isdigit(ch)) data = 10 * data + ch - '0', ch = getchar(); 
    return w * data; 
} 
const int MAX_N = 1e6 + 5; 
#define lson (o << 1) 
#define rson (o << 1 | 1) 
struct Node { long long ls, rs, ms, ss; } t[MAX_N << 2]; 
void pushup(int o) { 
	t[o].ms = t[lson].ms + t[rson].ms; 
	t[o].ls = max(t[lson].ms + t[rson].ls, t[lson].ls); 
	t[o].rs = max(t[rson].ms + t[lson].rs, t[rson].rs); 
	t[o].ss = max(max(t[lson].ss, t[rson].ss), t[lson].rs + t[rson].ls); 
} 
void modify(int o, int l, int r, int pos, int v) { 
	if (l == r) return (void)(t[o] = (Node){v, v, v, v}); 
	int mid = (l + r) >> 1; 
	if (pos <= mid) modify(lson, l, mid, pos, v); 
	else modify(rson, mid + 1, r, pos, v); 
	pushup(o); 
} 
int N, M, a[MAX_N], w[MAX_N]; 
int pre[MAX_N], lst[MAX_N]; 
int main () {
#ifndef ONLINE_JUDGE 
    freopen("cpp.in", "r", stdin);
#endif 
	N = gi(), M = gi(); 
	for (int i = 1; i <= N; i++) a[i] = gi(); 
	for (int i = 1; i <= M; i++) w[i] = gi(); 
	long long ans = 0; 
	for (int i = 1; i <= N; i++) { 
		pre[i] = lst[a[i]], lst[a[i]] = i; 
		if (pre[i]) modify(1, 1, N, pre[i], -w[a[i]]); 
		if (pre[pre[i]]) modify(1, 1, N, pre[pre[i]], 0); 
		modify(1, 1, N, i, w[a[i]]); 
		ans = max(ans, t[1].ss); 
	} 
	printf("%lld
", ans); 
    return 0; 
}