hdu_4897_Little Devil I(树链剖分)

题目连接：http://acm.hdu.edu.cn/showproblem.php?pid=4897

题意：有三种操作，1是在树上的两个节点之间的路径改变当前的颜色，2是改变树上有且只有一个端点在u,v之间的边的颜色，3是询问u,v之间黑色边的条数

题解：对于1，就是一般的树链剖分操作，对于2，我们知道树剖把树分成了重链和轻链，我们要充分利用这个结构才能降低时间复杂度，这里我们建立两个线段树来保存重链的sum和对每一个点的标记vis，第二个线段树表示如果这个点标记为1，那么以这个点周围的边都会改变，然后对于重链上的点，我们就直接在sum上操作，遇到轻链上的点，就在vis上标记，最后查询的时候先找sum上真实改变的边，然后当 当前的top跳到另一个top时，此时要查询vis上和sum上异或的值，最后全部加起来就是答案，要好好的想一想才能理解

 #include<cstdio>
 #define F(i,a,b) for(int i=a;i<=b;i++)

 const int N=1e5+;
 int t,n,op,q,x,y,g[N],nxt[N<<],v[N<<],ed;
 inline void adg(int x,int y){v[++ed]=y,nxt[ed]=g[x],g[x]=ed;}
 //--------------树链剖分---------------
 int dep[N],sz[N],hs[N],top[N],fa[N],tid[N],idx;
 void dfs1(int u,int pre){
     fa[u]=pre,sz[u]=,dep[u]=dep[pre]+,hs[u]=;
     for(int i=g[u];~i;i=nxt[i])if(v[i]!=pre)
     dfs1(v[i],u),sz[u]+=sz[v[i]],hs[u]=sz[v[i]]>sz[hs[u]]?v[i]:hs[u];
 }
 void dfs2(int u,int tp){
     tid[u]=++idx,top[u]=tp;
     if(hs[u])dfs2(hs[u],tp);
     for(int i=g[u];~i;i=nxt[i])
     if(v[i]!=fa[u]&&v[i]!=hs[u])dfs2(v[i],v[i]);
 }
 //--------------线段树------------------
 #define root 1,n,1
 #define ls l,m,rt<<1
 #define rs m+1,r,rt<<1|1
 int sum[N<<],vis[N<<],ly1[N<<];

 inline void sink(int rt,int tot){ly1[rt]^=,sum[rt]=tot-sum[rt];}

 inline void pd(int op,int l,int r,int rt){
     if(op==){
         if(ly1[rt]){
         int m=(l+r)>>;
         sink(rt<<,m-l+),sink(rt<<|,r-m);
         ly1[rt]=;
         }
     }else if(vis[rt])vis[rt<<]^=,vis[rt<<|]^=,vis[rt]=;
 }
 inline void up_p(int L,int R,int l,int r,int rt){
     if(L<=l&&r<=R){
         sink(rt,r+-l);
         return;
     }
     pd(,l,r,rt);
     int m=(l+r)>>;
     if(L<=m)up_p(L,R,ls);
     if(R>m)up_p(L,R,rs);
     sum[rt]=sum[rt<<]+sum[rt<<|];
 }
 inline int q_p(int L,int R,int l,int r,int rt){
     if(L<=l&&r<=R)return sum[rt];
     int m=(l+r)>>,ret=;
     pd(,l,r,rt);
     if(L<=m)ret+=q_p(L,R,ls);
     if(m<R)ret+=q_p(L,R,rs);
     return ret;
 }
 inline void up_v(int L,int R,int l,int r,int rt){
     if(L<=l&&r<=R){vis[rt]^=;return;}
     pd(,l,r,rt);
     int m=(l+r)>>;
     if(L<=m)up_v(L,R,ls);
     if(R>m)up_v(L,R,rs);
 }
 inline int q_v(int x,int l,int r,int rt){
     if(l==r)return vis[rt];
     pd(,l,r,rt);
     int m=(l+r)>>,ret=;
     if(x<=m)ret^=q_v(x,ls);
     else ret^=q_v(x,rs);
     return ret;
 }
 //--------------------------------------
 inline int ask(int x,int y){
     int ans=;
     while(top[x]!=top[y]){
         if(dep[top[x]]<dep[top[y]])x^=y,y^=x,x^=y;
         if(x!=top[x])ans+=q_p(tid[top[x]]+,tid[x],root);
         ans+=q_v(tid[fa[top[x]]],root)^q_p(tid[top[x]],tid[top[x]],root);
         x=fa[top[x]];
     }
     if(x==y)return ans;
     if(dep[x]>dep[y])x^=y,y^=x,x^=y;
     ans+=q_p(tid[x]+,tid[y],root);
     return ans;
 }
 inline void up(int op,int x,int y){
     if(op==){
         while(top[x]!=top[y]){
             if(dep[top[x]]<dep[top[y]])x^=y,y^=x,x^=y;
             up_p(tid[top[x]],tid[x],root),x=fa[top[x]];
         }
         if(x!=y){
             if(dep[x]>dep[y])x^=y,y^=x,x^=y;
             up_p(tid[x]+,tid[y],root);
         }
     }else{
         while(top[x]!=top[y]){
             if(dep[top[x]]<dep[top[y]])x^=y,y^=x,x^=y;
             up_v(tid[top[x]],tid[x],root);
             up_p(tid[top[x]],tid[top[x]],root);
             if(hs[x])up_p(tid[x]+,tid[hs[x]],root);
             x=fa[top[x]];
         }
             if(dep[x]>dep[y])x^=y,y^=x,x^=y;
             up_v(tid[x],tid[y],root);
             up_p(tid[x],tid[x],root);
             if(hs[y])up_p(tid[y]+,tid[hs[y]],root);
     }
 }

 int main(){
     scanf("%d",&t);
     while(t--){
         scanf("%d",&n);
         F(i,,n)g[i]=-;ed=;
         F(i,,n<<)sum[i]=,vis[i]=,ly1[i]=;
         F(i,,n-)scanf("%d%d",&x,&y),adg(x,y),adg(y,x);
         dfs1(,),idx=,dfs2(,),scanf("%d",&q);
         while(q--){
             scanf("%d%d%d",&op,&x,&y);
             if(op<)up(op,x,y);else printf("%d\n",ask(x,y));
         }
     }
     return ;
 }