[Bzoj4196] [NOI2015] 软件包管理器 [树链剖分，线段树]

题解摘要：树链剖分后用线段树区间查询修改，对于安装软件，将改点到根的路径全部变为1，对于卸载软件，将子树清空。注意边界，编号是从0开始的，容易漏掉树根。

第一次写树剖～

 #include <iostream>
 #include <algorithm>
 #include <cstdio>
 #include <cstdlib>
 #include <cstring>
 #include <cmath>
 #include <ctime>

 using namespace std;

 struct Edge
 {
     int    to,next;
 }e[];

 int    ind,n,Pre[],f[],Sum[],Out_dfn[];
 int    p[],cnt,hv[],Head[],Dfn[];

 void    Add_edge(const int & x,const int & y)
 {
     e[++cnt].to=y;
     e[cnt].next=p[x];
     p[x]=cnt;
     return ;
 }

 int    Init_Dfs(const int& S)
 {
     int    num=,i,Max_=;
     for(i=p[S];i;i=e[i].next)
     {
         int temp=Init_Dfs(e[i].to);
         if(temp>Max_)Max_=temp,hv[S]=e[i].to;
         num+=temp;
     }
     return num;
 }

 void    Dfs(const int& S,const int& fa)
 {
     int    i;
     if(!Head[S])Head[S]=Head[fa];Dfn[S]=++ind;

     if(hv[S])Dfs(hv[S],S);
     for(i=p[S];i;i=e[i].next)
     {
         if(e[i].to!=hv[S])
         {
             Head[e[i].to]=e[i].to;
             Pre[e[i].to]=S;
             Dfs(e[i].to,S);
         }
     }
     Out_dfn[S]=ind;
     return ;
 }

 inline void    push_down(const int & num,const int & l,const int & r)
 {
     if(f[num])
     {
         if(l!=r)
         {
             f[num<<]=f[num<<|]=f[num];
         }
         if(f[num]==)Sum[num]=r-l+;
         if(f[num]==-)Sum[num]=;
         f[num]=;
     }
     return ;
 }

 void    push_up(const int & num,const int & l,const int & r)
 {
     if(l!=r)
     {
         int    mid=l+((r-l)>>);
         push_down(num<<,l,mid);
         push_down(num<<|,mid+,r);
         Sum[num]=Sum[num<<]+Sum[num<<|];
     }
     return ;
 }

 void    Change(const int & l,const int & r,const int & num,
         const int & s,const int & t,const int &d)
 {
     if(s<=l && r<=t)
     {
         f[num]=d;
         push_down(num,l,r);
         return ;
     }
     int    mid=l+((r-l)>>);
     push_down(num,l,r);
     if(s<=mid)Change(l,mid,num<<,s,t,d);
     if(t>mid) Change(mid+,r,num<<|,s,t,d);
     push_up(num,l,r);
     return ;
 }

 int    Query(const int & l,const int & r,const int & num,
         const int & s,const int & t)
 {
     if(s<=l && r<=t)
     {
         push_down(num,l,r);
         return Sum[num];
     }

     int    mid=l+((r-l)>>),temp=;
     push_down(num,l,r);
     if(s<=mid)temp+=Query(l,mid,num<<,s,t);
     if(t>mid) temp+=Query(mid+,r,num<<|,s,t);
     push_up(num,l,r);
     return temp;
 }

 int    Get_sum_install(const int& S)
 {
     int    x=S,temp=,sum=;

     while(x)
     {
         sum+=Query(,n,,Dfn[Head[x]],Dfn[x]);
         Change(,n,,Dfn[Head[x]],Dfn[x],);
         temp+=Dfn[x]-Dfn[Head[x]]+;
         x=Pre[Head[x]];
     }

     return temp-sum;
 }

 int    Get_sum_remove(const int & S)
 {
     int    x=S,sum=;

     sum=Query(,n,,Dfn[x],Out_dfn[x]);
     Change(,n,,Dfn[x],Out_dfn[x],-);

     return sum;
 }

 int main()
 {

     int    i,x,q;
     char    op[];

     scanf("%d",&n);
     for(i=;i<=n;++i)
     {
         scanf("%d",&x);x++;
         Add_edge(x,i);
     }

     Init_Dfs();

     Head[]=;
     Dfs(,);
     scanf("%d",&q);
     for(i=;i<=q;++i)
     {
         scanf("%s%d",op,&x);x++;
         if(op[]=='i')
             printf("%d\n",Get_sum_install(x));
         else
             printf("%d\n",Get_sum_remove(x));
     }

     return ;
 }

不知道为什么，BZOJ TLE了，但是UOJ AC（求大神指教）？？