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步操作中改变安装状态的软件包数。
树链剖分,线段树维护区间已安装软件包个数。
对install操作统计x到根的路径上未安装个数并设为已安装
对uninstall操作统计以x为根的子树中已安装个数并设为未安装
#include<cstdio> #define N 100005 inline int read(){ int x=0,c=getchar(); while(c>'9'||c<'0')c=getchar(); while(c>='0'&&c<='9')x=x*10+c-'0',c=getchar(); return x; } int sz[N],pa[N],top[N],son[N],id[N],im[N],idp=1; int n,m,a; char s[16]; int ss[N*2],nx[N*2],sp=N; void f1(int w){ sz[w]=1; for(int i=nx[w];i;i=nx[i]){ int u=ss[i]; f1(u); sz[w]+=sz[u]; if(!son[w]||sz[son[w]]<sz[u])son[w]=u; } } void f2(int w,int tp){ id[w]=idp++; top[w]=tp; if(son[w])f2(son[w],tp); for(int i=nx[w];i;i=nx[i]){ int u=ss[i]; if(u==son[w])continue; f2(u,u); } im[w]=idp-1; } int c1[262144]; bool t0[262144]; bool t1[262144]; void down(int w,int L,int R){ if(t0[w]){ t0[w]=0; if(L<R){ t0[w+w]=t0[w+w+1]=1; t1[w+w]=t1[w+w+1]=0; c1[w+w]=c1[w+w+1]=0; } } if(t1[w]){ t1[w]=0; if(L<R){ t1[w+w]=t1[w+w+1]=1; t0[w+w]=t0[w+w+1]=0; c1[w+w]=c1[w+w+1]=(R+1-L)>>1; } } } int set0(int l,int r,int w=1,int L=1,int R=131072){ int M=L+R>>1,ans=0; down(w,L,R); if(L==l&&R==r){ ans=c1[w]; c1[w]=0; t0[w]=1; t1[w]=0; return ans; } if(r<=M)ans+=set0(l,r,w+w,L,M); else if(l>M)ans+=set0(l,r,w+w+1,M+1,R); else ans+=set0(l,M,w+w,L,M)+set0(M+1,r,w+w+1,M+1,R); c1[w]=c1[w+w]+c1[w+w+1]; return ans; } int set1(int l,int r,int w=1,int L=1,int R=131072){ int M=L+R>>1,ans=0; down(w,L,R); if(L==l&&R==r){ ans=R+1-L-c1[w]; c1[w]=R+1-L; t0[w]=0; t1[w]=1; return ans; } if(r<=M)ans+=set1(l,r,w+w,L,M); else if(l>M)ans+=set1(l,r,w+w+1,M+1,R); else ans+=set1(l,M,w+w,L,M)+set1(M+1,r,w+w+1,M+1,R); c1[w]=c1[w+w]+c1[w+w+1]; return ans; } int main(){ n=read(); for(int i=2;i<=n;i++){ a=pa[i]=read()+1; ss[sp]=i; nx[sp]=nx[a]; nx[a]=sp++; } f1(1); f2(1,1); m=read(); while(m--){ scanf("%s",s); a=read()+1; if(s[0]=='i'){ int ans=0; int t=top[a]; while(a){ ans+=set1(id[t],id[a]); a=pa[t];t=top[a]; } printf("%d\n",ans); }else{ printf("%d\n",set0(id[a],im[a])); } } return 0; }