[CodeVS4633][Mz]树链剖分练习

思路：

轻重链剖分+线段树。

 #include<cstdio>
 #include<vector>
 #include<cstring>
 const int N=;
 std::vector<int> e[N];
 inline void add_edge(const int u,const int v) {
     e[u].push_back(v);
     e[v].push_back(u);
 }
 int par[N]={,},dep[N]={},size[N]={};
 void dfs1(const int x) {
     size[x]=;
     for(unsigned int i=;i<e[x].size();i++) {
         if(e[x][i]==par[x]) continue;
         par[e[x][i]]=x,dep[e[x][i]]=dep[x]+;
         dfs1(e[x][i]);
         size[x]+=size[e[x][i]];
     }
 }
 int top[N]={},dfn[N],sz=;
 void dfs2(const int x) {
     if(!top[x]) top[x]=x;
     dfn[x]=++sz;
     if(e[x].size()==&&x!=) return;
     int v=;
     for(unsigned int i=;i<e[x].size();i++) {
         if(e[x][i]==par[x]) continue;
         if(size[e[x][i]]>size[v]) v=e[x][i];
     }
     top[v]=top[x];
     dfs2(v);
     for(unsigned int i=;i<e[x].size();i++) {
         if(e[x][i]==par[x]||e[x][i]==v) continue;
         dfs2(e[x][i]);
     }
 }
 class SegmentTree {
     #define _left <<1
     #define _right <<1|1
     private:
         int val[N<<],tag[N<<];
         int len(const int l,const int r) {
             return r-l+;
         }
         void push_down(const int p,const int b,const int e) {
             if(!tag[p]) return;
             int mid=(b+e)>>;
             tag[p _left]+=tag[p];
             tag[p _right]+=tag[p];
             val[p _left]+=tag[p]*len(b,mid);
             val[p _right]+=tag[p]*len(mid+,e);
             tag[p]=;
         }
         void push_up(const int p) {
             val[p]=val[p _left]+val[p _right];
         }
     public:
         SegmentTree() {
             memset(val,,sizeof val);
             memset(tag,,sizeof tag);
         }
         void modify(const int p,const int b,const int e,const int l,const int r) {
             if((b==l)&&(e==r)) {
                 val[p]+=len(b,e);
                 tag[p]++;
                 return;
             }
             push_down(p,b,e);
             int mid=(b+e)>>;
             if(l<=mid) modify(p _left,b,mid,l,std::min(mid,r));
             if(r>mid) modify(p _right,mid+,e,std::max(mid+,l),r);
             push_up(p);
         }
         int query(const int p,const int b,const int e,const int l,const int r) {
             if((b==l)&&(e==r)) return val[p];
             push_down(p,b,e);
             int mid=(b+e)>>,ans=;
             if(l<=mid) ans+=query(p _left,b,mid,l,std::min(mid,r));
             if(r>mid) ans+=query(p _right,mid+,e,std::max(mid+,l),r);
             return ans;
         }
 };
 SegmentTree t;
 inline void swap(int &x,int &y) {
     int t;
     t=x;
     x=y;
     y=t;
 }
 int n;
 void modify(int x,int y) {
     for(;top[x]!=top[y];x=par[top[x]]) {
         if(dep[top[x]]<dep[top[y]]) swap(x,y);
         t.modify(,,n,dfn[top[x]],dfn[x]);
     }
     if(dep[x]<dep[y]) swap(x,y);
     t.modify(,,n,dfn[y],dfn[x]);
 }
 int query(int x,int y) {
     int ans=;
     for(;top[x]!=top[y];x=par[top[x]]) {
         if(dep[top[x]]<dep[top[y]]) swap(x,y);
         ans+=t.query(,,n,dfn[top[x]],dfn[x]);
     }
     if(dep[x]<dep[y]) swap(x,y);
     ans+=t.query(,,n,dfn[y],dfn[x]);
     return ans;
 }
 int main() {
     scanf("%d",&n);
     for(int i=;i<n;i++) {
         int x,y;
         scanf("%d%d",&x,&y);
         add_edge(x,y);
     }
     dfs1();
     dfs2();
     int q;
     scanf("%d",&q);
     while(q--) {
         int op,x,y;
         scanf("%d%d%d",&op,&x,&y);
         if(op==) modify(x,y);
         if(op==) printf("%d\n",query(x,y));
     }
     return ;
 }