BZOJ4196: [Noi2015]软件包管理器（树链剖分）

Description

Linux用户和OSX用户一定对软件包管理器不会陌生。通过软件包管理器，你可以通过一行命令安装某一个软件包，然后软件包管理器会帮助你从软件源下载软件包，同时自动解决所有的依赖（即下载安装这个软件包的安装所依赖的其它软件包），完成所有的配置。Debian/Ubuntu使用的apt-get，Fedora/CentOS使用的yum，以及OSX下可用的homebrew都是优秀的软件包管理器。

你决定设计你自己的软件包管理器。不可避免地，你要解决软件包之间的依赖问题。如果软件包A依赖软件包B，那么安装软件包A以前，必须先安装软件包B。同时，如果想要卸载软件包B，则必须卸载软件包A。现在你已经获得了所有的软件包之间的依赖关系。而且，由于你之前的工作，除0号软件包以外，在你的管理器当中的软件包都会依赖一个且仅一个软件包，而0号软件包不依赖任何一个软件包。依赖关系不存在环（若有m(m≥2)个软件包A1,A2,A3,…,Am，其中A1依赖A2，A2依赖A3，A3依赖A4，……，Am−1依赖Am，而Am依赖A1，则称这m个软件包的依赖关系构成环），当然也不会有一个软件包依赖自己。

现在你要为你的软件包管理器写一个依赖解决程序。根据反馈，用户希望在安装和卸载某个软件包时，快速地知道这个操作实际上会改变多少个软件包的安装状态（即安装操作会安装多少个未安装的软件包，或卸载操作会卸载多少个已安装的软件包），你的任务就是实现这个部分。注意，安装一个已安装的软件包，或卸载一个未安装的软件包，都不会改变任何软件包的安装状态，即在此情况下，改变安装状态的软件包数为0。

Input

输入文件的第1行包含1个正整数n，表示软件包的总数。软件包从0开始编号。

随后一行包含n−1个整数，相邻整数之间用单个空格隔开，分别表示1,2,3,…,n−2,n−1号软件包依赖的软件包的编号。

接下来一行包含1个正整数q，表示询问的总数。

之后q行，每行1个询问。询问分为两种：

installx：表示安装软件包x

uninstallx：表示卸载软件包x

你需要维护每个软件包的安装状态，一开始所有的软件包都处于未安装状态。对于每个操作，你需要输出这步操作会改变多少个软件包的安装状态，随后应用这个操作（即改变你维护的安装状态）。

Output

输出文件包括q行。

输出文件的第i行输出1个整数，为第i步操作中改变安装状态的软件包数。

 

Sample Input

7
0 0 0 1 1 5
5
install 5
install 6
uninstall 1
install 4
uninstall 0

Sample Output

3
1
3
2
3

解题思路：

显然这是一棵树。

支持询问到根节点有多少点被标记，清空子树标记。

一颗线段树，支持区间改为1/0。

维护树链剖分序即可。

然而被卡常，开了O2才过，好蒟蒻啊QAQ

代码：

 #pragma GCC optimize(2)
 #include<cstdio>
 #include<cstring>
 #include<algorithm>
 #define lll spc<<1
 #define rrr spc<<1|1
 inline int read()
 {
     int l=;
     int ret=;
     char ch=getchar();
     while(ch<''||ch>'')
     {
         if(ch=='-')
             l=-l;
         ch=getchar();
     }
     while(ch<=''&&ch>='')
     {
         ret=ret*+ch-'';
         ch=getchar();
     }
     return ret*l;
 }
 struct trnt{
     int val;
     int len;
     int lzt;
     bool chg;
 }tr[];
 struct pnt{
     int hd;
     int fa;
     int dp;
     int tp;
     int ind;
     int mxs;
     int wgt;
 }p[];
 struct ent{
     int twd;
     int lst;
 }e[];
 int cnt;
 int n,m;
 int dfn;
 char cmd[];
 void ade(int f,int t)
 {
     cnt++;
     e[cnt].twd=t;
     e[cnt].lst=p[f].hd;
     p[f].hd=cnt;
 }
 void pushup(int spc)
 {
     tr[spc].val=tr[lll].val+tr[rrr].val;
     tr[spc].len=tr[lll].len+tr[rrr].len;
     return ;
 }
 void Add(int spc,int x)
 {
     tr[spc].val=tr[spc].len*x;
     tr[spc].lzt=x;
     tr[spc].chg=;
     return ;
 }
 void pushdown(int spc)
 {
     if(tr[spc].chg)
     {
         Add(lll,tr[spc].lzt);
         Add(rrr,tr[spc].lzt);
         tr[spc].chg=;
     }
     return ;
 }
 void build(int l,int r,int spc)
 {
     if(l==r)
     {
         tr[spc].len=;
         return ;
     }
     int mid=(l+r)>>;
     build(l,mid,lll);
     build(mid+,r,rrr);
     pushup(spc);
     return ;
 }
 void update(int l,int r,int ll,int rr,int spc,int v)
 {
     if(l>rr||ll>r)
         return ;
     if(ll<=l&&r<=rr)
     {
         Add(spc,v);
         return ;
     }
     pushdown(spc);
     int mid=(l+r)>>;
     update(l,mid,ll,rr,lll,v);
     update(mid+,r,ll,rr,rrr,v);
     pushup(spc);
     return ;
 }
 int query(int l,int r,int ll,int rr,int spc)
 {
     if(ll>r||l>rr)
         return ;
     if(ll<=l&&r<=rr)
         return tr[spc].val;
     pushdown(spc);
     int mid=(l+r)>>;
     return query(l,mid,ll,rr,lll)+query(mid+,r,ll,rr,rrr);
 }
 void Basic_dfs(int x,int f)
 {
     p[x].fa=f;
     p[x].dp=p[f].dp+;
     p[x].wgt=;
     int maxs=-;
     for(int i=p[x].hd;i;i=e[i].lst)
     {
         int to=e[i].twd;
         if(to==f)
             continue;
         Basic_dfs(to,x);
         p[x].wgt+=p[to].wgt;
         if(maxs<p[to].wgt)
         {
             maxs=p[to].wgt;
             p[x].mxs=to;
         }
     }
     return ;
 }
 void Build_dfs(int x,int top)
 {
     if(!x)
         return ;
     p[x].ind=++dfn;
     p[x].tp=top;
     Build_dfs(p[x].mxs,top);
     for(int i=p[x].hd;i;i=e[i].lst)
     {
         int to=e[i].twd;
         if(p[to].ind)
             continue;
         Build_dfs(to,to);
     }
 }
 int Install(int x)
 {
     int ans=;
     while(p[x].tp!=)
     {
         ans+=p[x].dp-p[p[x].tp].dp-query(,dfn,p[p[x].tp].ind,p[x].ind,)+;
         update(,dfn,p[p[x].tp].ind,p[x].ind,,);
         x=p[p[x].tp].fa;
     }
     ans+=p[x].dp-query(,dfn,p[].ind,p[x].ind,);
     update(,dfn,p[].ind,p[x].ind,,);
     return ans;
 }
 int Uninstall(int x)
 {
     int ans=query(,dfn,p[x].ind,p[x].ind+p[x].wgt-,);
     update(,dfn,p[x].ind,p[x].ind+p[x].wgt-,,);
     return ans;
 }
 int main()
 {
     n=read();
     for(int i=;i<=n;i++)
     {
         int x=read();
         x++;
         ade(x,i);
         ade(i,x);
     }
     dfn=;
     Basic_dfs(,);
     Build_dfs(,);
     build(,dfn,);
     m=read();
     while(m--)
     {
         scanf("%s",cmd+);
         int x=read();
         x++;
         if(cmd[]=='i')
             printf("%d\n",Install(x));
         else
             printf("%d\n",Uninstall(x));
     }
     return ;
 }