bzoj4196 [Noi2015]软件包管理器 树链剖分+线段树

先把树剖分了（又是dfs1、dfs2），然后区间求和、区间覆盖即可

难得的1A好（shui）题

——写了那么多题，终于有一道是1A的了，加上上一次连续交了几遍A的程序，我的状态莫名好看啊233

总结：

1.开始用define简化代码

2.写的时候出现了手抖，还不够熟练

3.剖分基本上没有问题，线段树题目欠多

 #include <cstdio>
 #include <iostream>
 #define lson now*2
 #define rson now*2+1
 #define mid (l+r)/2
 using namespace std;
 int cnt=,n,m;
 int flag[],t[];
 int fa[],size[],son[],bro[],b[],pos[],top[];
 int dfs1(int k)
 {
     size[k]=;
     for(int i=son[k];i!=;i=bro[i])
         size[k]+=dfs1(i);
     return size[k];
 }
 void dfs2(int k,int to)
 {
     b[++cnt]=k;
     top[k]=to;
     int max=son[k];
     pos[k]=cnt;
     for(int i=son[k];i!=;i=bro[i])
         if(size[i]>size[max])
             max=i;
     if(max==)
         return;
     dfs2(max,to);
     for(int i=son[k];i!=;i=bro[i])
         if(i!=max)
             dfs2(i,i);
 }
 void down(int now,int l,int r)
 {
     if(flag[now]==)
     {
         flag[lson]=;
         flag[rson]=;
         t[lson]=mid-l+;
         t[rson]=r-mid;
         flag[now]=;
     }
     if(flag[now]==)
     {
         flag[lson]=;
         flag[rson]=;
         t[lson]=;
         t[rson]=;
         flag[now]=;
     }
 }
 int que(int now,int l,int r,int x,int y)
 {
     if(l==x && r==y) return t[now];
     down(now,l,r);
     return ((x<=mid)?que(lson,l,mid,x,min(y,mid)):)+((y>mid)?que(rson,mid+,r,max(x,mid+),y):);
 }
 void set_0(int now,int l,int r,int x,int y)
 {
     if(l==x && r==y)
     {
         flag[now]=;
         t[now]=;
         return;
     }
     down(now,l,r);
     if(x<=mid)
         set_0(lson,l,mid,x,min(y,mid));
     if(y>mid)
         set_0(rson,mid+,r,max(x,mid+),y);
     t[now]=t[lson]+t[rson];
 }
 void set_1(int now,int l,int r,int x,int y)
 {
     if(l==x && r==y)
     {
         flag[now]=;
         t[now]=r-l+;
         return;
     }
     down(now,l,r);
     if(x<=mid)
         set_1(lson,l,mid,x,min(y,mid));
     if(y>mid)
         set_1(rson,mid+,r,max(x,mid+),y);
     t[now]=t[lson]+t[rson];
 }
 int main()
 {
     scanf("%d",&n);
     for(int i=;i<=n;i++)
     {
         scanf("%d",&fa[i]);
         fa[i]++;
         bro[i]=son[fa[i]];
         son[fa[i]]=i;
     }
     dfs1();
     dfs2(,);
     scanf("%d",&m);
     for(int i=;i<=m;i++)
     {
         char ch=getchar();
         while(ch!='i' && ch!='u')
             ch=getchar();
         bool b=ch=='i';
         int x=;
         while(ch<'' || ch>'')
             ch=getchar();
         while(ch>='' && ch<='')
         {
             x=x*+ch-'';
             ch=getchar();
         }
         x++;
         if(b)
         {
             int sum=;
             for(int i=top[x];;i=top[x])
             {
                 sum+=pos[x]-pos[i]+-que(,,n,pos[i],pos[x]);
                 set_1(,,n,pos[i],pos[x]);
                 if(i==)
                     break;
                 x=fa[i];
             }
             printf("%d\n",sum);
         }
         else
         {
             printf("%d\n",que(,,n,pos[x],pos[x]+size[x]-));
             set_0(,,n,pos[x],pos[x]+size[x]-);
         }
     }
     return ;
 }