BZOJ 1984月下“毛景树” LCT维护边权 + 下传标记

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; 
}