题意:
维护一棵树, 需要满足修改路径上的值和询问路径上最大连续子段和。
思路:
用lct维护。 修改直接搞个lazy就好, 询问的时候如果是求出lca的话比较麻烦, 但是可以先把u弄成树的跟, 然后access(v), 然后在splay(u), 这时候u节点所在的splay树里面就是u到v的所有节点, 所以输出最大值就可以了。。
重点还是在push_up和push_down。。。 其他都是模板。。。
#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define N 100020
#define M 200020
#define inf 0x3f3f3f3f
int pre[N], ch[N][2], key[N], sz[N];
int sum[N], lx[N], rx[N], mx[N];
int c[N];
bool rev[N], rt[N];
int n;
int fst[N], nxt[M], vv[M], e;
void init() {
for(int i = 0; i <= n; ++i) {
fst[i] = -1;
rev[i] = false;
rt[i] = true;
ch[i][0] = ch[i][1] = 0;
pre[i] = 0;
if(i) sz[i] = 1;
c[i] = -inf;
}
e = 0;
}
void add(int u, int v) {
vv[e] = v, nxt[e] = fst[u], fst[u] = e++;
}
void debug() {
for(int i = 1; i <= n; ++i) {
printf("node[%d]:pre %d ch %d %d sz %d rev %s\n", i, pre[i], ch[i][0], ch[i][1], sz[i], rev[i]? "true": "false");
}
}
void dfs(int u, int p) {
pre[u] = p;
for(int i = fst[u]; ~i; i = nxt[i]) {
int v = vv[i];
if(v == p) continue;
dfs(v, u);
}
}
void push_up(int x) {
int lc = ch[x][0], rc = ch[x][1];
sum[x] = sum[lc] + sum[rc] + key[x];
sz[x] = sz[lc] + sz[rc] + 1;
if(lc == 0 && rc == 0) {
mx[x] = lx[x] = rx[x] = key[x];
return;
}
if(rc == 0) {
mx[x] = max(key[x], mx[lc]);
mx[x] = max(mx[x], key[x] + rx[lc]);
lx[x] = max(lx[lc], sum[lc] + key[x]);
rx[x] = max(key[x], key[x] + rx[lc]);
return ;
}
if(lc == 0) {
mx[x] = max(key[x], mx[rc]);
mx[x] = max(mx[x], key[x] + lx[rc]);
lx[x] = max(key[x], key[x] + lx[rc]);
rx[x] = max(rx[rc], sum[rc] + key[x]);
return;
}
mx[x] = max(mx[lc], mx[rc]);
mx[x] = max(mx[x], key[x]);
mx[x] = max(mx[x], key[x] + rx[lc]);
mx[x] = max(mx[x], key[x] + lx[rc]);
mx[x] = max(mx[x], key[x] + rx[lc] + lx[rc]);
lx[x] = max(lx[lc], sum[lc] + key[x]);
lx[x] = max(lx[x], sum[lc] + key[x] + lx[rc]);
rx[x] = max(rx[rc], sum[rc] + key[x]);
rx[x] = max(rx[x], sum[rc] + key[x] + rx[lc]);
}
void update_c(int x, int val) {
if(!x) return;
key[x] = val;
sum[x] = val * sz[x];
c[x] = val;
if(val > 0) {
mx[x] = lx[x] = rx[x] = sum[x];
}
else
mx[x] = lx[x] = rx[x] = val;
}
void update_rev(int x) {
if(!x) return;
swap(ch[x][0], ch[x][1]);
rev[x] ^= 1;
swap(lx[x], rx[x]);
}
void push_down(int x) {
if(c[x] != -inf) {
update_c(ch[x][0], c[x]);
update_c(ch[x][1], c[x]);
c[x] = -inf;
}
if(rev[x]) {
update_rev(ch[x][0]);
update_rev(ch[x][1]);
rev[x] = false;
}
}
void rot(int x) {
int y = pre[x], d = ch[y][1] == x;
ch[y][d] = ch[x][!d];
pre[ch[x][!d]] = y;
ch[x][!d] = y;
pre[x] = pre[y];
pre[y] = x;
if(rt[y]) rt[y] = false, rt[x] = true;
else {
ch[pre[x]][ch[pre[x]][1]==y] = x;
}
push_up(y);
}
void P(int x) {
if(!rt[x]) P(pre[x]);
push_down(x);
}
void splay(int x) {
P(x);
while(!rt[x]) {
int f = pre[x], ff = pre[f];
if(rt[f]) rot(x);
else if((ch[ff][1] == f) == (ch[f][1] == x))
rot(f), rot(x);
else rot(x), rot(x);
}
push_up(x);
}
int access(int x) {
int y = 0;
for(; x; y = x, x = pre[x]) {
splay(x);
rt[ch[x][1]] = true;
ch[x][1] = y;
rt[y] = false;
push_up(x);
}
return y;
}
void lca(int &u, int &v) {
access(v), v = 0;
for(splay(u); pre[u]; v = u, u = pre[u], splay(u)) {
rt[ch[u][1]] = true;
ch[u][1] = v;
rt[v] = false;
push_up(u);
}
}
void make_root(int x) {
access(x);
splay(x);
update_rev(x);
}
int main() {
//freopen("tt.txt", "r", stdin);
while(scanf("%d", &n) != EOF) {
init();
for(int i = 1; i <= n; ++i) {
scanf("%d", &key[i]);
sum[i] = mx[i] = lx[i] = rx[i] = key[i];
}
for(int i = 1; i < n; ++i) {
int u, v;
scanf("%d%d", &u, &v);
add(u, v), add(v, u);
}
dfs(1, 0);
int m;
scanf("%d", &m);
while(m--) {
int op, u, v, c;
scanf("%d%d%d", &op, &u, &v);
if(op == 2) scanf("%d", &c);
if(op == 2) {
lca(u, v);
key[u] = c;
update_c(ch[u][1], c);
update_c(v, c);
push_up(u);
}
else {
make_root(u);
access(v);
splay(u);
//debug();
printf("%d\n", max(0, mx[u]));
}
}
}
return 0;
}