BZOJ 4196: [Noi2015]软件包管理器 树链剖分

4196: [Noi2015]软件包管理器

Time Limit: 10 Sec  Memory Limit: 512 MB
Submit: 1752  Solved: 1006
[Submit][Status][Discuss]

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

HINT

 一开始所有的软件包都处于未安装状态。

安装 5 号软件包，需要安装 0,1,5 三个软件包。

之后安装 6 号软件包，只需要安装 6 号软件包。此时安装了 0,1,5,6 四个软件包。

卸载 1 号软件包需要卸载 1,5,6 三个软件包。此时只有 0 号软件包还处于安装状态。

之后安装 4 号软件包，需要安装 1,4 两个软件包。此时 0,1,4 处在安装状态。

最后，卸载 0 号软件包会卸载所有的软件包。

n=100000

q=100000

树链剖分

log^2n安装 logn卸载

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<complex>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<bitset>
#include<queue>
#include<set>
#include<map>

using namespace std;

inline int read()
{
	register int x=0,f=1;register char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=(x<<1)+(x<<3)+ch-'0';ch=getchar();}
	return x*f;
}
inline void print(int x)
{if(x<0)x=-x,putchar('-');if(x>=10)print(x/10);putchar(x%10+'0');}

const int N=100100;

int ecnt,last[N],n;
struct EDGE{int to,nt;}e[N<<1];
inline void add(int u,int v)
{e[++ecnt]=(EDGE){v,last[u]};last[u]=ecnt;}

int fa[N],size[N],deep[N];

void dfs1(int u)
{
	size[u]=1;register int i;
	for(i=last[u];i;i=e[i].nt)
	{dfs1(e[i].to);size[u]+=size[e[i].to];deep[e[i].to]=deep[u]+1;fa[e[i].to]=u;}
}

int pos[N],dfn[N],maxd[N],top[N],cnt=0;

void dfs2(int u,int tp)
{
	int k=0;top[u]=tp;
	dfn[u]=maxd[u]=++cnt;pos[cnt]=u;
	for(int i=last[u];i;i=e[i].nt)
		if(size[k]<size[e[i].to])k=e[i].to;
	if(!k)return ;dfs2(k,tp);
	for(int i=last[u];i;i=e[i].nt)
	{
		if(e[i].to!=k)dfs2(e[i].to,e[i].to);
		maxd[u]=max(maxd[u],maxd[e[i].to]);
	}
}

struct seg_tree{int sum0,sum1,l,r,mark;}tr[N*5];

inline void pushup(int k)
{tr[k].sum0=tr[k<<1].sum0+tr[k<<1|1].sum0;tr[k].sum1=tr[k<<1].sum1+tr[k<<1|1].sum1;}

inline void pushdown(int k)
{
	if(tr[k].mark==1)
	{
		tr[k].mark=0;
		tr[k<<1].mark=tr[k<<1|1].mark=1;
		tr[k<<1].sum0=tr[k<<1].r-tr[k<<1].l+1;
		tr[k<<1|1].sum0=tr[k<<1|1].r-tr[k<<1|1].l+1;
		tr[k<<1].sum1=0;
		tr[k<<1|1].sum1=0;
	}
	if(tr[k].mark==2)
	{
		tr[k].mark=0;                 
		tr[k<<1].mark=tr[k<<1|1].mark=2;
		tr[k<<1].sum1=tr[k<<1].r-tr[k<<1].l+1;
		tr[k<<1|1].sum1=tr[k<<1|1].r-tr[k<<1|1].l+1;
		tr[k<<1].sum0=0;
		tr[k<<1|1].sum0=0;
	}
}

void build(int k,int l,int r)
{
	tr[k].l=l;tr[k].r=r;
	if(l==r){tr[k].sum0=1;return ;}
	int mid=l+r>>1;
	build(k<<1,l,mid);build(k<<1|1,mid+1,r);
	pushup(k);
}

int query_1(int k,int x,int y)
{
	int l=tr[k].l,r=tr[k].r;
	if(l>=x&&r<=y)return tr[k].sum1;
	int mid=l+r>>1;pushdown(k);
	if(mid<x)return query_1(k<<1|1,x,y);
	else if(mid>=y)return query_1(k<<1,x,y);
	else return query_1(k<<1,x,y)+query_1(k<<1|1,x,y);
}

int query_0(int k,int x,int y)
{
	int l=tr[k].l,r=tr[k].r;
	if(l>=x&&r<=y)return tr[k].sum0;
	int mid=l+r>>1;pushdown(k);
	if(mid<x)return query_0(k<<1|1,x,y);
	else if(mid>=y)return query_0(k<<1,x,y);
	else return query_0(k<<1,x,y)+query_0(k<<1|1,x,y);
}

void modify(int k,int x,int y,int val)
{
	int l=tr[k].l,r=tr[k].r;
	if(l>=x&&r<=y)
	{
		if(val)tr[k].sum1=r-l+1,tr[k].sum0=0;
		else tr[k].sum1=0,tr[k].sum0=r-l+1;
		tr[k].mark=val+1;
		return ;
	}
	pushdown(k);
	int mid=l+r>>1;
	if(mid<x)modify(k<<1|1,x,y,val);
	else if(mid>=y)modify(k<<1,x,y,val);
	else modify(k<<1,x,y,val),modify(k<<1|1,x,y,val);
	pushup(k);
}

inline void install(int x)
{
	register int u=x,v=1,ans=0;
	while(top[u]!=top[v])
	{
		if(deep[top[u]]<deep[top[v]])swap(u,v);
		ans+=query_0(1,dfn[top[u]],dfn[u]);
		modify(1,dfn[top[u]],dfn[u],1);u=fa[top[u]];
	}
	if(deep[u]>deep[v])swap(u,v);
	ans+=query_0(1,dfn[u],dfn[v]);
	modify(1,dfn[u],dfn[v],1);
	print(ans);puts("");
}

inline void uninstall(int x)
{
	register int u=x,ans=0;
	ans=query_1(1,dfn[u],maxd[u]);
	modify(1,dfn[u],maxd[u],0);
	print(ans);puts("");
}

void initial()
{dfs1(1);dfs2(1,1);build(1,1,n);}

int main()
{
	n=read();
	register int i,u,x;
	for(i=1;i<n;++i)u=read()+1,add(u,i+1);
	initial();int Q=read();char ch;
	while(Q--)
	{
		ch=getchar();while(ch^'i'&&ch^'u')ch=getchar();
		if(ch=='i'){x=read()+1;install(x);}
		else {x=read()+1;uninstall(x);}
	}
	return 0;
}