NOI2015软件包管理器 树剖线段树

题目：

一棵树，兹磁

　　1.查询根到一个点的染色点数并全染好

　　2.查询子树内染色点数并把颜色洗掉

树剖裸题，丝毫不虚（为什么我考试的时候碰不到这种好题呢）好像20min写完搞定

 #include <bits/stdc++.h>
 #define mid (l+r>>1)
 using namespace std;
 int TIME,n,m,t;char ch;
 int size[],fir[],nex[],top[];
 int pos[],en[],fa[];
 int fz[],sum[];
 int dfs(int now)
 {
     size[now]=;
     for(int i=fir[now];i;i=nex[i])
         size[now]+=dfs(i);
     return size[now];
 }
 void pou(int now,int tp)
 {
     top[now]=tp;pos[now]=++TIME;
     int best=fir[now];
     for(int i=fir[now];i;i=nex[i])
         if(size[i]>size[best]) best=i;
     if(best)
         pou(best,tp);
     for(int i=fir[now];i;i=nex[i])
         if(i!=best) pou(i,i);
     en[now]=TIME;
 }
 void down(int now,int l,int r)
 {
     if(fz[now]>-)
     {
         fz[now<<]=fz[now<<|]=fz[now];
         sum[now<<]=(mid-l+)*fz[now];
         sum[now<<|]=(r-mid)*fz[now];
         fz[now]=-;
     }
 }
 int que(int now,int l,int r,int x,int y)
 {
     if(l==x && r==y)
         return sum[now];
     down(now,l,r);
     int ret=;
     if(x<=mid)
         ret+=que(now<<,l,mid,x,min(mid,y));
     if(y>mid)
         ret+=que(now<<|,mid+,r,max(mid+,x),y);
     return ret;
 }
 void change(int now,int l,int r,int x,int y,int z)
 {
     if(l==x && r==y)
     {
         sum[now]=z*(r-l+);
         fz[now]=z;
         return;
     }
     down(now,l,r);
     if(x<=mid)
         change(now<<,l,mid,x,min(mid,y),z);
     if(y>mid)
         change(now<<|,mid+,r,max(mid+,x),y,z);
     sum[now]=sum[now<<]+sum[now<<|];
 }
 int main()
 {
     scanf("%d",&n);
     for(int i=;i<=n;i++)
         scanf("%d",&fa[i]),++fa[i],nex[i]=fir[fa[i]],fir[fa[i]]=i;
     dfs();pou(,);
     scanf("%d",&m);
     for(int i=;i<=m;i++)
     {
         for(ch=getchar();!isalpha(ch);ch=getchar());
         if(ch=='i')
         {
             for(int i=;i<=;i++)
                 getchar();
             scanf("%d",&t);++t;
             int ret=;
             for(;t;t=fa[top[t]])
                 ret+=pos[t]-pos[top[t]]+-que(,,n,pos[top[t]],pos[t]),
                 change(,,n,pos[top[t]],pos[t],);
             printf("%d\n",ret);
         }
         if(ch=='u')
         {
             for(int i=;i<=;i++)
                 getchar();
             scanf("%d",&t);++t;
             printf("%d\n",que(,,n,pos[t],en[t]));
             change(,,n,pos[t],en[t],);
         }
     }
     return ;
 }