[POI2015]KIN[线段树]
很套路的维护最大和子段
#include <cmath>
#include <cstring>
#include <cstdio>
#include <cstdlib>
#include <algorithm>
#include <complex>
#include <ctime>
#include <vector>
#define mp(x, y) make_pair(x, y)
using namespace std;
const int N = 1e6 + 5;
int n, m, L;
long long ans, res;
int a[N], b[N];
int pre[N], rec[N];
bool vis[N];
struct Seg{
long long w[N << 2], ls[N << 2], rs[N << 2], sum[N << 2];
void update(int rt){
sum[rt] = sum[rt << 1] + sum[rt << 1 | 1];
w[rt] = rs[rt << 1] + ls[rt << 1 | 1];
ls[rt] = max(ls[rt << 1], sum[rt << 1] + ls[rt << 1 | 1]);
rs[rt] = max(rs[rt << 1 | 1], sum[rt << 1 | 1] + rs[rt << 1]);
// printf("%d %lld %lld %lld\n", rt, w[rt], ls[rt], rs[rt]);
}
void mdf(int rt, int l, int r, int x, int d){
if(l == r){sum[rt] = d, ls[rt] = rs[rt] = w[rt] = max(0, d); return;}
int mid = l + ((r - l) >> 1);
if(x <= mid) mdf(rt << 1, l, mid, x, d);
else mdf(rt << 1 | 1, mid + 1, r, x, d);
update(rt);
}
}seg;
int main(){
scanf("%d%d", &n, &m);
for(int i = 1; i <= n; ++i)
scanf("%d", &a[i]), pre[i] = rec[a[i]], rec[a[i]] = i;
for(int i = 1; i <= m; ++i) scanf("%d", &b[i]);
for(int i = 1; i <= n; ++i){
seg.mdf(1, 1, n, i, b[a[i]]);
if(pre[i]) seg.mdf(1, 1, n, pre[i], -b[a[i]]);
if(pre[pre[i]]) seg.mdf(1, 1, n, pre[pre[i]], 0);
ans = max(ans, seg.w[1]);
}
printf("%lld\n", ans);
return 0;
}