HYSBZ 1036 树的统计Count 树链剖分

这题就是裸的树链剖分

传送门：http://www.lydsy.com/JudgeOnline/problem.php?id=1036

#include <map>
#include <set>
#include <queue>
#include <cmath>
#include <cstdio>
#include <cstring>
#include <cstdlib>
#include <iostream>
#include <algorithm>

using namespace std;

const int MAXN = 100010;

struct Edge {
    int to ;
    Edge * next;
}E[MAXN*2],*EE;

struct Gragh {
    Edge * first;
}G[MAXN];

struct Tree {
    int x,y;
    int sum,max_num;
}t[MAXN<<2];

int pre[MAXN],son[MAXN],siz[MAXN],dep[MAXN],top[MAXN],pos[MAXN],Rank[MAXN];
int N,Q,S;
int num[MAXN];
int idx[MAXN];
int tree_idx = 0;
int tot ;

void Init() {
    EE = E;
    dep[1] = 1;
    pre[1] = 0;
    tree_idx = 0;
    memset(G,0,sizeof(G));
    memset(son,0,sizeof(son));
}

// void addedge(int u,int v,int value,int id) {
//     EE->to = v ; EE->next = G[u].first ; EE->value = value ; EE->id = id ; G[u].first = EE++;
//     EE->to = u ; EE->next = G[v].first ; EE->value = value ; EE->id = id ; G[v].first = EE++;
// }

void addedge(int u,int v) {
    EE->to = v ; EE->next = G[u].first ; G[u].first = EE++;
    EE->to = u ; EE->next = G[v].first ; G[v].first = EE++;
}

void dfs1(int u) {
    siz[u] = 1;
    son[u] = 0;
    for(Edge * p = G[u].first ; p ; p = p -> next) {
        if(p->to != pre[u]) {
            pre[p->to] = u;
            dep[p->to] = dep[u] + 1;
            dfs1(p->to);
            siz[u] += siz[p->to];
            if(siz[p->to] > siz[son[u]])
                son[u] = p -> to;
        }
    }
}

void dfs2(int u,int ancestor) {
    top[u] = ancestor;
    tree_idx ++;
    pos[u] = tree_idx;
    if(son[u] != 0) {
        dfs2(son[u],ancestor);
    }
    for(Edge * p = G[u].first ; p ; p = p -> next) {
        if(p->to != pre[u] && p->to != son[u]) {
            dfs2(p->to,p->to);
            // num[pos[p->to]] = p->value;
            // idx[p->id] = pos[p->to];
        }
        // else if(p->to == son[u]) {
        //     num[pos[p->to]] = p->value;
        //     idx[p->id] = pos[p->to];
        // }
    }
} 

void Push_Up(int rt) {
    t[rt].max_num = max(t[rt<<1].max_num,t[rt<<1|1].max_num);
    t[rt].sum = t[rt<<1].sum + t[rt<<1|1].sum;
}

void Build(int x,int y,int rt) {
    t[rt].x = x ; t[rt].y = y;
    if(x == y) {
        t[rt].max_num = Rank[x];
        t[rt].sum = Rank[x];
        return ;
    }
    int mid = (x + y) >> 1;
    Build(x,mid,rt<<1);
    Build(mid+1,y,rt<<1|1);
    Push_Up(rt);
}

void Update(int rt,int position,int value) {
   if(t[rt].x == t[rt].y) {
        t[rt].max_num = value;
        t[rt].sum = value;
        return ;
    }
    int mid = (t[rt].x + t[rt].y) >> 1;
    if(mid >= position) {
        Update(rt<<1,position,value);
    }
    else {
        Update(rt<<1|1,position,value);
    }
    Push_Up(rt);
}

int max_num,sum;

void Query(int rt,int left,int right) {
    if(left <= t[rt].x && right >= t[rt].y) {
        max_num = max(max_num,t[rt].max_num);
        sum = sum + t[rt].sum;
        return ;
    }
    int mid = (t[rt].x + t[rt].y) >> 1;
    if(mid >= left) {
        Query(rt<<1,left,right);
    } 
    if(mid < right){
        Query(rt<<1|1,left,right);
    }
    Push_Up(rt);
}

void Query_Tree(int x,int y) {
    while(top[x] != top[y]) {
        if(dep[top[x]] < dep[top[y]]) swap(x,y);
        Query(1,pos[top[x]],pos[x]);
        x = pre[top[x]];
    }
    if(dep[x] > dep[y]) swap(x,y);
    Query(1,pos[x],pos[y]);
}

void input() {
    int u,v,value;
    for(int i = 1 ; i <= N - 1 ; i++) {
        scanf("%d %d",&u,&v);
        addedge(u,v);
    } 
    for(int i = 1 ; i <= N ; i++) {
        scanf("%d",num+i);
    }
    dfs1(1);
    dfs2(1,1);
    for(int i = 1 ; i <= N ; i++) {
        Rank[pos[i]] = num[i];
    }
    Build(1,N,1);
    char op[10];
    int color;
    scanf("%d",&Q);
    for(int i = 1 ; i <= Q ; i ++) {
        scanf("%s",op) ;
        scanf("%d %d",&u,&v);
        if(strcmp(op,"CHANGE")==0) {
            Update(1,pos[u],v);
            //Change(u,v,color);
        }
        else {
            max_num = -2e9; sum = 0;
            Query_Tree(u,v);
            if(strcmp(op,"QSUM")==0) {
                printf("%d\n",sum);
            }
            else {
                printf("%d\n",max_num);
            }

        }
        //print_tree(1);
        //puts("");
    }
}

int main(void) {
    freopen("a.in","r",stdin);
    while(~scanf("%d",&N)) {
        Init();
        input();
        //solve();
    }
    return 0;
}