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+7;
int t,n,op,q,x,y,g[N],nxt[N<<1],v[N<<1],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]=1,dep[u]=dep[pre]+1,hs[u]=0;
    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<<2],vis[N<<2],ly1[N<<2];

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

inline void pd(int op,int l,int r,int rt){
    if(op==1){
		if(ly1[rt]){
        int m=(l+r)>>1;
		sink(rt<<1,m-l+1),sink(rt<<1|1,r-m);
        ly1[rt]=0;
		}
	}else if(vis[rt])vis[rt<<1]^=1,vis[rt<<1|1]^=1,vis[rt]=0;
}
inline void up_p(int L,int R,int l,int r,int rt){
    if(L<=l&&r<=R){
        sink(rt,r+1-l);
        return;
    }
    pd(1,l,r,rt);
    int m=(l+r)>>1;
    if(L<=m)up_p(L,R,ls);
    if(R>m)up_p(L,R,rs);
    sum[rt]=sum[rt<<1]+sum[rt<<1|1];
}
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)>>1,ret=0;
    pd(1,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]^=1;return;}
    pd(2,l,r,rt);
    int m=(l+r)>>1;
    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(2,l,r,rt);
    int m=(l+r)>>1,ret=0;
	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=0;
    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]]+1,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]+1,tid[y],root);
    return ans;
}
inline void up(int op,int x,int y){
    if(op==1){
        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]+1,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]+1,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]+1,tid[hs[y]],root);
    }
}

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