bzoj1036 zjoi2008 树的统计 count

填坑= =第一道裸树剖

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cstdlib>
#include<cmath>
#include<vector>

using namespace std;

const int Maxn=,Maxm=,INF=~0u>>;

int n;

struct Data{
    int sum,max;
    Data (int sum=,int max=-INF):sum(sum),max(max){}
    Data operator + (const Data& rhs)const{
        return Data(sum+rhs.sum,std::max(max,rhs.max));
    }
};

struct SegmentTree{
    Data da[Maxn*];
    int lft,rgt,w;
    void pp(int l,int r=){
        lft=l;rgt=r;
    }
    void change(int s,int l,int r){
        if(l==r && l==lft){
            da[s]=Data(w,w);
            return;
        }
        int mid=(l+r)>>;
        if(lft<=mid) change(s<<,l,mid);
        else change(s<<|,mid+,r);
        da[s]=da[s<<]+da[s<<|];
    }
    Data query(int s,int l,int r){
        if(lft<=l && r<=rgt) return da[s];
        int mid=(l+r)>>;
        if(rgt<=mid) return query(s<<,l,mid);
        if(mid<lft) return query(s<<|,mid+,r);
        return query(s<<,l,mid) + query(s<<|,mid+,r);
    }
}seg;

int en[Maxm*],next[Maxm*],fir[Maxn],tot;

void Add(int from,int to){
    en[++tot] = to;
    next[tot] = fir[from];
    fir[from] = tot;
}

int sz[Maxn],fa[Maxn],son[Maxn],dep[Maxn];

void dfs(int u){
    sz[u]=;
    son[u]=;
    for(int k=fir[u];k;k=next[k]){
        int v=en[k];
        if(v == fa[u]) continue;
        fa[v] = u;
        dep[v]=dep[u]+;
        dfs(v);
        sz[u]+=sz[v];
        if(sz[v]>sz[son[u]]) son[u] = v;
    }
}

int pos[Maxn],top[Maxn],tm;

void build(int u,int pre){
    top[u]=pre;
    pos[u]=++tm;
    if(son[u]) build(son[u],pre);
    for(int k=fir[u];k;k=next[k]){
        int v=en[k];
        if(v==fa[u] || v==son[u]) continue;
        build(v,v);
    }
}

Data query(int a,int b){
    int ta=top[a],tb=top[b];
    Data ret;
    for(;ta!=tb;a=fa[ta],ta=top[a]){
        if(dep[ta]<dep[tb]) swap(a,b), swap(ta,tb);
        seg.pp(pos[ta],pos[a]);
        ret=ret+seg.query(,,n);
    }
    if(dep[a]>dep[b]) swap(a,b);
    seg.pp(pos[a],pos[b]);
    return ret+seg.query(,,n);
}

int main(){
#ifdef DEBUG
    freopen("in.txt","r",stdin);
    freopen("my.out","w",stdout);
#endif

    scanf("%d",&n);
    for(int i=;i<n;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        Add(u,v);
        Add(v,u);
    }

    dfs();
    build(,);

    for(int i=;i<=n;i++){
        seg.pp(pos[i]);
        scanf("%d",&seg.w);
        seg.change(,,n);
    }

    int m;
    scanf("%d",&m);
    char cmd[];
    int a,b;

    for(;m--;){
        scanf("%s%d%d",cmd,&a,&b);
        if(cmd[]=='C') seg.lft=pos[a],seg.w=b,seg.change(,,n);
        else if(cmd[]=='S') printf("%d\n",query(a,b).sum);
        else printf("%d\n",query(a,b).max);
    }

    return ;
}