hdu 4010 Query on The Trees（动态树）

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=300010;
int ch[maxn][2],pre[maxn],key[maxn];
int add[maxn],rev[maxn],Max[maxn];
bool rt[maxn];
void update_add(int r,int d){
  if(!r) return;
  key[r]+=d;
  add[r]+=d;
  Max[r]+=d;
}
void update_rev(int r){
  if(!r) return;
  swap(ch[r][0],ch[r][1]);
  rev[r]^=1;
}
void pushdown(int r){
   if(add[r]){
    update_add(ch[r][0],add[r]);
    update_add(ch[r][1],add[r]);
    add[r]=0;
   }
   if(rev[r]){
    update_rev(ch[r][0]);
    update_rev(ch[r][1]);
    rev[r]=0;
   }
}
void pushup(int r){
    Max[r]=max(max(Max[ch[r][0]],Max[ch[r][1]]),key[r]);
}
void Rotate(int x){
   int y=pre[x],kind=ch[y][1]==x;
   ch[y][kind]=ch[x][!kind];
   pre[ch[y][kind]]=y;
   pre[x]=pre[y];
   pre[y]=x;
   ch[x][!kind]=y;
   if(rt[y]){
    rt[y]=false;rt[x]=true;
   }
   else{
    ch[pre[x]][ch[pre[x]][1]==y]=x;
   }
   pushup(y);
}
void P(int r){
  if(!rt[r]) P(pre[r]);
  pushdown(r);
}
void splay(int r){
  P(r);
  while(!rt[r]){
    int f=pre[r],ff=pre[f];
    if(rt[f]) Rotate(r);
    else if((ch[ff][1]==f)==(ch[f][1]==r))
        Rotate(f),Rotate(r);
    else
        Rotate(r),Rotate(r);
  }
  pushup(r);
}
int Access(int x){  //构造重链，用splay维护
   int y=0;
   for(;x;x=pre[y=x]){
    splay(x);
    rt[ch[x][1]]=true;rt[ch[x][1]=y]=false;
    pushup(x);
   }
   return y;
}
bool judge(int u,int v){
   while(pre[u]) u=pre[u];
   while(pre[v]) v=pre[v];
   return u==v;
}
void mroot(int r){
   Access(r);
   splay(r);
   update_rev(r);
}
void lca(int &u,int &v)
{
    Access(v),v=0;
    while(u){
        splay(u);
        if(!pre[u]) return;
        rt[ch[u][1]]=true;
        rt[ch[u][1]=v]=false;
        pushup(u);
        u=pre[v=u];
    }
}
void link(int u,int v){  //合并
    if(judge(u,v)){
        puts("-1");return;
    }
    mroot(u);
    pre[u]=v;
}
void cut(int u,int v){   //分离
   if(u==v||!judge(u,v)){
    puts("-1");return;
   }
   mroot(u);
   splay(v);
   pre[ch[v][0]]=pre[v];
   pre[v]=0;
   rt[ch[v][0]]=true;
   ch[v][0]=0;
   pushup(v);
}
void ADD(int u,int v,int w){
   if(!judge(u,v)){
     puts("-1");
     return;
   }
   lca(u,v);
   update_add(ch[u][1],w);
   update_add(v,w);
   key[u]+=w;
   pushup(u);
}
void query(int u,int v){
  if(!judge(u,v)){
    puts("-1");return;
  }
  lca(u,v);
  printf("%d\n",max(max(Max[v],Max[ch[u][1]]),key[u]));
}
struct Edge{
  int to,next;
}edge[maxn*2];
int head[maxn],tot;
void addedge(int u,int v){
  edge[tot].to=v;
  edge[tot].next=head[u];
  head[u]=tot++;
}
void dfs(int u){
  for(int i=head[u];i!=-1;i=edge[i].next){
    int v=edge[i].to;
    if(pre[v]!=0) continue;
    pre[v]=u;
    dfs(v);
  }
}
int main()
{
    int n,q,u,v;
    while(scanf("%d",&n)!=EOF){
        tot=0;
        for(int i=0;i<=n;i++){
            head[i]=-1;
            pre[i]=0;
            ch[i][0]=ch[i][1]=0;
            rev[i]=0;
            add[i]=0;
            rt[i]=true;
        }
        Max[0]=-0x3f3f3f3f;
        for(int i=1;i<n;i++){
            scanf("%d%d",&u,&v);
            addedge(u,v);
            addedge(v,u);
        }
        for(int i=1;i<=n;i++){
            scanf("%d",&key[i]);
            Max[i]=key[i];
        }
        scanf("%d",&q);
        pre[1]=-1;
        dfs(1); //处理出pre
        pre[1]=0;
        int op;
        while(q--){
            scanf("%d",&op);
            if(op==1){
                int x,y;
                scanf("%d%d",&x,&y);
                link(x,y);
            }
            else if(op==2){
                int x,y;
                scanf("%d%d",&x,&y);
                cut(x,y);
            }
            else if(op==3){
                int w,x,y;
                scanf("%d%d%d",&w,&x,&y);
                ADD(x,y,w);
            }
            else{
                int x,y;
                scanf("%d%d",&x,&y);
                query(x,y);
            }
        }
        printf("\n");
    }
    return 0;
}