【BZOJ4196】[NOI2015] 软件包管理器（树链剖分）

点此看题面

大致题意： 有\(n\)个软件包，它们的依赖关系形成一棵树。现在，问你安装或卸载一个软件包，会影响多少个软件包的安装状态。

树链剖分

这道题应该是 树链剖分 算法比较入门的题目吧。

对于安装操作

我们对安装和卸载两种操作分别处理。

首先，让我们来看一看安装操作应该如何实现。

考虑到要安装一个软件包，就要把它所依赖的软件包，它的依赖所依赖的软件包，一直到\(0\)号软件包，全部安装。

如果将题目中给出的关系看成一棵树，那么也就是要把这个软件包到\(0\)号节点（即 根节点 ）的路径上的所有软件包全部安装。

用树链剖分应该可以轻松做到这一点。

对于卸载操作

我们再来看看如何实现卸载操作。

考虑卸掉一个软件包，那么所有依赖于它的软件包将全被卸载，所有依赖于依赖于它的软件包的软件包，也将全被卸载，以此类推。

不难想到，这就相当于把以这个软件包为根的子树内的所有软件包给卸载了。

用树链剖分应该也能很轻松做到这一点。

代码

#include<bits/stdc++.h>
#define max(x,y) ((x)>(y)?(x):(y))
#define min(x,y) ((x)<(y)?(x):(y))
#define abs(x) ((x)<0?-(x):(x))
#define LL long long
#define ull unsigned long long
#define swap(x,y) (x^=y,y^=x,x^=y)
#define N 100000
#define add(x,y) (e[++ee].nxt=lnk[x],e[lnk[x]=ee].to=y)
using namespace std;
int n,ee=0,lnk[N+5];
struct edge
{
	int to,nxt;
}e[2*N+5];
class FIO
{
	private:
		#define Fsize 100000
		#define tc() (FinNow==FinEnd&&(FinEnd=(FinNow=Fin)+fread(Fin,1,Fsize,stdin),FinNow==FinEnd)?EOF:*FinNow++)
		#define pc(ch) (FoutSize<Fsize?Fout[FoutSize++]=ch:(fwrite(Fout,1,FoutSize,stdout),Fout[(FoutSize=0)++]=ch))
		int f,FoutSize,OutputTop;char ch,Fin[Fsize],*FinNow,*FinEnd,Fout[Fsize],OutputStack[Fsize];double w;
	public:
		FIO() {FinNow=FinEnd=Fin;}
		inline void read(int &x) {x=0,f=1;while(!isdigit(ch=tc())) f=ch^'-'?1:-1;while(x=(x<<3)+(x<<1)+(ch&15),isdigit(ch=tc()));x*=f;}
		inline void read_char(char &x) {while(isspace(x=tc()));}
		inline void read_string(string &x) {x="";while(isspace(ch=tc()));while(x+=ch,!isspace(ch=tc()));}
		inline void write(int x) {if(!x) return (void)pc('0');if(x<0) pc('-'),x=-x;while(x) OutputStack[++OutputTop]=x%10+48,x/=10;while(OutputTop) pc(OutputStack[OutputTop]),--OutputTop;}
		inline void write_char(char x) {pc(x);}
		inline void write_string(string x) {register int i,len=x.length();for(i=0;i<len;++i) pc(x[i]);}
		inline void end() {fwrite(Fout,1,FoutSize,stdout);}
}F;
class TreeChainDissection//树链剖分模板
{
	private:
		#define PushUp(x) (Sum[x]=Sum[x<<1]+Sum[x<<1|1])
		#define PushDown(x,ln,rn) ((~flag[x]?(flag[x]?(Sum[x<<1]=ln,Sum[x<<1|1]=rn,flag[x<<1]=flag[x<<1|1]=1):(Sum[x<<1]=Sum[x<<1|1]=flag[x<<1]=flag[x<<1|1]=0)):0),flag[x]=-1)
		int d,fa[N+5],son[N+5],sz[N+5],Top[N+5],dfn[N+5],fac[N+5],Depth[N+5],Sum[N<<2],flag[N<<2];
		inline void dfs1(int x)//第一遍dfs预处理
		{
			register int i;
			for(sz[x]=1,i=lnk[x];i;i=e[i].nxt)
			{
				if(!(fa[x]^e[i].to)) continue;
				fa[e[i].to]=x,Depth[e[i].to]=Depth[x]+1,dfs1(e[i].to),sz[x]+=sz[e[i].to];
				if(sz[e[i].to]>sz[son[x]]) son[x]=e[i].to; 
			}
		}
		inline void dfs2(int x,int col)//第二遍dfs预处理
		{
			register int i;
			if(son[fac[dfn[x]=++d]=x]) dfs2(son[x],col);
			for(Top[x]=col,i=lnk[x];i;i=e[i].nxt)
			{
				if(!(fa[x]^e[i].to&&son[x]^e[i].to)) continue;
				dfs2(e[i].to,e[i].to);
			}
		}
		inline void Build(int l,int r,int rt)//建树
		{
			flag[rt]=-1;
			if(l^r)
			{
				register int mid=l+r>>1;
				if(l<=mid) Build(l,mid,rt<<1);
				if(r>mid) Build(mid+1,r,rt<<1|1);
			}
		}
		inline int Update(int l,int r,int rt,int ul,int ur,int x)//将[ul...ur]区间内所有软件包的安装状态修改为x（x=0表示未安装，x=1表示安装），并返回需要改变安装状态的软件包数
		{
			register int res=0;
			if(ul<=l&&r<=ur) {register int t;(x?(t=r-l+1,flag[rt]=1):(t=0,flag[rt]=0)),res=abs(t-Sum[rt]),Sum[rt]=t;return res;}
			register int mid=l+r>>1;
			PushDown(rt,mid-l+1,r-mid);
			if(ul<=mid) res=Update(l,mid,rt<<1,ul,ur,x);
			if(ur>mid) res+=Update(mid+1,r,rt<<1|1,ul,ur,x);
			PushUp(rt);
			return res;
		}
	public:
		inline void Init() {dfs1(1),dfs2(1,1),Build(1,n,1);}
		inline int uninstall(int pos) {return Update(1,n,1,dfn[pos],dfn[pos]+sz[pos]-1,0);}//卸载一个软件包，相当于修改以这个软件包为根节点的子树内所有软件包安装状态为0
		inline int install(int pos)//安装一个软件包，相当于修改这个软件包到根节点的所有软件包安装状态为1
		{
			register int res=0;
			while(Top[pos]>>1) res+=Update(1,n,1,dfn[Top[pos]],dfn[pos],1),pos=fa[Top[pos]];
			return res+Update(1,n,1,1,dfn[pos],1);
		}
}S;
int main()
{
	register int i,Q,x;register char op;
	for(F.read(n),i=2;i<=n;++i) F.read(x),add(x+1,i);
	for(S.Init(),F.read(Q);Q;--Q)
	{
		F.read_char(op),F.read(x);
		if(op^'u') F.write(S.install(x+1)),F.write_char('\n');//对于安装操作
		else F.write(S.uninstall(x+1)),F.write_char('\n');//对于卸载操作
	}
	return F.end(),0;
}