【NOI 2015】软件包管理器

【题目】

传送门

题目描述：

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

输入格式：

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

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

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

之后 q 行，每行 1 个询问。询问分为两种：
install x：表示安装软件包 x
uninstall x：表示卸载软件包 x

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

输出格式：

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

样例数据：

【样例1】

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

输出
3
1
3
2
3

【样例2】

输入
10
0 1 2 1 3 0 0 3 2
10
install 0
install 3
uninstall 2
install 7
install 5
install 9
uninstall 9
install 4
install 1
install 9

输出
1
3
2
1
3
1
1
1
0
1

备注：

【样例1说明】

一开始所有的软件包都处于未安装状态。
安装 5 号软件包，需要安装 0,1,5 三个软件包。
之后安装 6 号软件包，只需要安装 6 号软件包。此时安装了 0,1,5,6 四个软件包。
卸载 1 号软件包需要卸载 1,5,6 三个软件包。此时只有 0 号软件包还处于安装状态。
之后安装 4 号软件包，需要安装 1,4 两个软件包。此时 0,1,4 处在安装状态。
最后，卸载 0 号软件包会卸载所有的软件包。

【数据范围】

【分析】

题目大意：给出一棵有根树（根为 $0$），树上的点有点权（一开始都是 $0$），给出以下两个操作：

1. install x：输出 x 到根的路径上 0 的个数，并将路径上所有点的点权改为 1；
2. uninstall x：输出 x 的子树（包括 x）中 1 的个数，并将子树中所有点的点权改为 0。

将题意简化了之后，是不是发现这就是一道树剖板题！

直接树剖，然后用线段树来维护就可以了。

Ps：由于我自己的代码习惯，代码中我就直接把根设为 1 了。

【代码】

#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 100005
using namespace std;
int n,q,t,tot;char s[10];
int first[N],v[N],next[N];
int dep[N],pos[N],son[N],top[N],size[N],father[N];
int sum[N<<2],add[N<<2],mark[N<<2];
void edge(int x,int y)
{
	t++;
	next[t]=first[x];
	first[x]=t;
	v[t]=y;
}
void point(int x,int y)
{
	tot++;
	top[x]=y;
	pos[x]=tot;
}
void dfs1(int x)
{
	int i,j;
	size[x]=1;
	for(i=first[x];i;i=next[i])
	{
		j=v[i];
		father[j]=x;
		dep[j]=dep[x]+1;
		dfs1(j);
		size[x]+=size[j];
		if(size[son[x]]<size[j])
		  son[x]=j;
	}
}
void dfs2(int x)
{
	if(son[x])
	{
		point(son[x],top[x]);
		dfs2(son[x]);
	}
	int i,j;
	for(i=first[x];i;i=next[i])
	{
		j=v[i];
		if(j!=son[x])
		{
			point(j,j);
			dfs2(j);
		}
	}
}
void pushdown(int root,int l,int r,int mid)
{
	sum[root<<1]=add[root]*(mid-l+1);
	sum[root<<1|1]=add[root]*(r-mid);
	add[root<<1]=add[root<<1|1]=add[root];
	mark[root]=0,mark[root<<1]=mark[root<<1|1]=1;
}
void modify(int root,int l,int r,int x,int y,int k)
{
	if(l>=x&&r<=y)
	{
		add[root]=k;
		mark[root]=1;
		sum[root]=k*(r-l+1);
		return;
	}
	int mid=(l+r)>>1;
	if(mark[root])  pushdown(root,l,r,mid);
	if(x<=mid)  modify(root<<1,l,mid,x,y,k);
	if(y>mid)  modify(root<<1|1,mid+1,r,x,y,k);
	sum[root]=sum[root<<1]+sum[root<<1|1];
}
int query(int root,int l,int r,int x,int y)
{
	if(l>=x&&r<=y)
	  return sum[root];
	int ans=0,mid=(l+r)>>1;
	if(mark[root])  pushdown(root,l,r,mid);
	if(x<=mid)  ans+=query(root<<1,l,mid,x,y);
	if(y>mid)  ans+=query(root<<1|1,mid+1,r,x,y);
	return ans;
}
int install(int x)
{
	int ans=dep[x]+1;
	while(top[x]!=1)
	{
		ans-=query(1,1,n,pos[top[x]],pos[x]);
		modify(1,1,n,pos[top[x]],pos[x],1);
		x=father[top[x]];
	}
	ans-=query(1,1,n,1,pos[x]);
	modify(1,1,n,1,pos[x],1);
	return ans;
}
int uninstall(int x)
{
	int ans=query(1,1,n,pos[x],pos[x]+size[x]-1);
	modify(1,1,n,pos[x],pos[x]+size[x]-1,0);
	return ans;
}
int main()
{
	int x,i;
	scanf("%d",&n);
	for(i=2;i<=n;++i)
	{
		scanf("%d",&x);
		edge(x+1,i);
	}
	point(1,1);
	dfs1(1);dfs2(1);
	scanf("%d",&q);
	for(i=1;i<=q;++i)
	{
		scanf("%s%d",s+1,&x);x++;
		if(s[1]=='i')  printf("%d\n",install(x));
		if(s[1]=='u')  printf("%d\n",uninstall(x));
	}
	return 0;
}