树剖裸题——BZOJ1036 树的统计

 #include<cstring>
 #include<cmath>
 #include<algorithm>
 #include<cstdio>
 #define foru(i,x,y) for(int i=x;i<=y;i++)
 #define clr(a) memset(a,0,sizeof(a))
 using namespace std;
 const int N=;
 struct edge{int to,nxt;}e[N*];
 struct node{int s,m;}t[*N];
 int d[N],id[N],head[N],f[N],siz[N],son[N],top[N];
 //f[v]        节点v的父节点编号
 //id[v]        节点v的父边在线段树中的编号
 //siz[v]    以节点v为根的数中的节点数
 //son[v]    节点v的子节点中siz[]最大的节点编号
 //top[v]    节点v所在重链的顶端节点编号
 //d[v]        节点v的深度
 int ne,cnt,n;

 void add(int a,int b){
     e[++ne]=(edge){b,head[a]};head[a]=ne;
 }
 void dfs(int k,int fa,int dep){//统计f[] siz[] son[] d[]
     //printf("test\n");
     f[k]=fa;d[k]=dep;siz[k]=;son[k]=;
     for(int i=head[k];i;i=e[i].nxt){
         int v=e[i].to;
         if(v==fa)continue;
         dfs(v,k,dep+);
         siz[k]+=siz[v];
         if(siz[v]>siz[son[k]])son[k]=v;
     }
 }

 void build(int k,int tp){
     id[k]=++cnt; top[k]=tp;//按序将边加入线段树
     if(son[k])build(son[k],tp);//重儿子的top[]从重链顶端继承
     for(int i=head[k];i;i=e[i].nxt){
         int v=e[i].to;
         if(v!=son[k]&&v!=f[k])
             build(v,v);//轻儿子top[]为自身
     }
 }

 #define mid ((L+R)>>1)
 #define ls (k<<1)//写位运算一定要开-Wall，否则一定要记得加括号
 #define rs ls+1

 void update(int k,int L,int R,int p,int x){
     if(p>R||p<L)return;
     if(L==R){t[k].s=t[k].m=x;return;}
     update(ls,L,mid,p,x); update(rs,mid+,R,p,x);
     t[k].m=max(t[ls].m,t[rs].m);
     t[k].s=t[ls].s+t[rs].s;
 }

 int querym(int k,int L,int R,int l,int r){
     if(l>R||r<L)return -1e9;
     if(l<=L&&R<=r)return t[k].m;
     return max(querym(ls,L,mid,l,r),querym(rs,mid+,R,l,r));
 }

 int querys(int k,int L,int R,int l,int r){
     if(l>R||r<L)return ;
     if(l<=L&&R<=r)return t[k].s;
     return querys(ls,L,mid,l,r)+querys(rs,mid+,R,l,r);
 }

 int findm(int x,int y){
     int ans=-1e9*;
     while(top[x]!=top[y]){//类似LCA，每次将较低的节点上跳，并统计路径上的最大值
         if(d[top[x]]<d[top[y]])swap(x,y);
         ans=max(ans,querym(,,cnt,id[top[x]],id[x]));
         x=f[top[x]];
     }
     if(d[x]>d[y])swap(x,y);//当两点处于同一条链上的时候，进行最后一次统计
     ans=max(ans,querym(,,cnt,id[x],id[y]));
     return ans;
 }

 int finds(int x,int y){
     int ans=;
     while(top[x]!=top[y]){
         if(d[top[x]]<d[top[y]])swap(x,y);
         ans+=querys(,,cnt,id[top[x]],id[x]);
         x=f[top[x]];
     }
     if(d[x]>d[y])swap(x,y);
     ans+=querys(,,cnt,id[x],id[y]);
     return ans;
 }
 char ch[];
 int main(){
     int u,v,x,y;
     scanf("%d",&n);
     foru(i,,n-){
         scanf("%d%d",&u,&v);
         add(u,v);add(v,u);
     }
     dfs(,,);
     build(,);
     foru(i,,n){
         scanf("%d",&u);
         update(,,cnt,id[i],u);
     }
     scanf("%d",&u);
     while(u--){
         scanf("%s%d%d",ch,&x,&y);
         if(ch[]=='C')update(,,cnt,id[x],y);
         else{
             if(ch[]=='M')printf("%d\n",findm(x,y));
             else printf("%d\n",finds(x,y));
         }
     }
     return ;
 }