Description
毛毛虫经过及时的变形,最终逃过的一劫,离开了菜妈的菜园。 毛毛虫经过千山万水,历尽千辛万苦,最后来到了小小的绍兴一中的校园里。爬啊爬~爬啊爬~~毛毛虫爬到了一颗小小的“毛景树”下面,发现树上长着他最爱吃的毛毛果~~~ “毛景树”上有N个节点和N-1条树枝,但节点上是没有毛毛果的,毛毛果都是长在树枝上的。但是这棵“毛景树”有着神奇的魔力,他能改变树枝上毛毛果的个数: Change k w:将第k条树枝上毛毛果的个数改变为w个。 Cover u v w:将节点u与节点v之间的树枝上毛毛果的个数都改变为w个。 Add u v w:将节点u与节点v之间的树枝上毛毛果的个数都增加w个。 由于毛毛虫很贪,于是他会有如下询问: Max u v:询问节点u与节点v之间树枝上毛毛果个数最多有多少个。
Input
第一行一个正整数N。 接下来N-1行,每行三个正整数Ui,Vi和Wi,第i+1行描述第i条树枝。表示第i条树枝连接节点Ui和节点Vi,树枝上有Wi个毛毛果。 接下来是操作和询问,以“Stop”结束。
Output
对于毛毛虫的每个询问操作,输出一个答案。
题解:树剖太简单了,练一练 LCT
#include <bits/stdc++.h>
#define setIO(s) freopen(s".in","r",stdin)
#define maxn 100050
#define inf 0x3f
using namespace std;
int n;
char str[10];
namespace LCT
{
#define lson ch[x][0]
#define rson ch[x][1]
int maxv[maxn << 1], tag[maxn << 1], rev[maxn << 1], addv[maxn << 1], val[maxn << 1], siz[maxn << 1], f[maxn << 1], sta[maxn << 1], ch[maxn << 1][2]; int isRoot(int x)
{
return !(ch[f[x]][0] == x || ch[f[x]][1] == x);
}
int get(int x)
{
return ch[f[x]][1] == x;
}
void mark_tag(int x, int delta)
{
if(!x) return;
addv[x] = 0;
tag[x] = (siz[x] ? delta : 0);
maxv[x] = (siz[x] ? delta : 0);
if(x > n) val[x] = delta;
}
void mark_add(int x, int delta)
{
if(!x) return;
addv[x] += (siz[x] ? delta : 0);
maxv[x] += (siz[x] ? delta : 0);
if(x > n) val[x] += delta;
}
void mark_rev(int x)
{
if(!x) return;
swap(lson, rson), rev[x] ^= 1;
}
void pushup(int x)
{
maxv[x] = max(val[x], max(maxv[lson], maxv[rson]));
siz[x] = (x > n) + siz[lson] + siz[rson];
}
void pushdown(int x)
{
if(tag[x]) mark_tag(lson, tag[x]), mark_tag(rson, tag[x]), tag[x] = 0;
if(addv[x]) mark_add(lson, addv[x]), mark_add(rson, addv[x]), addv[x] = 0;
if(rev[x]) mark_rev(lson), mark_rev(rson), rev[x] = 0;
}
void rotate(int x)
{
int old = f[x], fold = f[old], which = get(x);
if(!isRoot(old)) ch[fold][ch[fold][1] == old] = x;
ch[old][which] = ch[x][which ^ 1], f[ch[old][which]] = old;
ch[x][which ^ 1] = old, f[old] = x, f[x] = fold;
pushup(old), pushup(x);
}
void splay(int x)
{
int u = x, v = 0;
sta[++v] = u;
while(!isRoot(u)) sta[++v] = f[u], u = f[u];
while(v) pushdown(sta[v--]);
u = f[u];
for(int fa; (fa = f[x]) != u; rotate(x))
if(f[fa] != u) rotate(get(fa) == get(x) ? fa : x);
}
void Access(int x)
{
int y = 0;
while(x) splay(x), rson = y, pushup(x), y = x, x = f[x];
}
void MakeRoot(int x)
{
Access(x), splay(x), mark_rev(x);
}
// y is new root
void split(int x, int y)
{
MakeRoot(x), Access(y), splay(y);
}
void link(int x, int y)
{
MakeRoot(y), f[y] = x;
}
};
int main()
{
// setIO("input");
LCT :: maxv[0] = -inf;
scanf("%d",&n);
for(int i = 1; i < n; ++i)
{
int u, v, w;
scanf("%d%d%d",&u,&v,&w);
LCT :: val[i + n] = w;
LCT :: link(u, i + n);
LCT :: link(i + n, v);
}
while(1)
{
scanf("%s",str);
int u, v, k, w;
if(str[0] == 'A')
{
scanf("%d%d%d",&u,&v,&w), LCT :: split(u, v), LCT :: mark_add(v, w);
}
if(str[0] == 'M')
{
scanf("%d%d",&u,&v), LCT :: split(u, v), printf("%d\n",LCT :: maxv[v]);
}
if(str[1] == 'o')
{
scanf("%d%d%d",&u,&v,&w), LCT :: split(u, v), LCT :: mark_tag(v, w);
}
if(str[1] == 'h')
{
scanf("%d%d",&k,&w);
LCT :: Access(k + n), LCT :: splay(k + n), LCT :: val[k + n] = w, LCT :: pushup(k + n);
}
if(str[0] == 'S') break;
}
return 0;
}