来自FallDream的博客,未经允许,请勿转载,谢谢。
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。
把安装看成1,未安装看成0
发现安装操作就是查询它到根上0的数量 并把他们都变成1
卸载操作就是查询子树内1的数量,并把他们都变成0
所以树剖之后线段树就行了
#include<iostream> #include<cstdio> #define MN 100000 using namespace std; inline int read() { int x = 0 , f = 1; char ch = getchar(); while(ch < '0' || ch > '9'){ if(ch == '-') f = -1; ch = getchar();} while(ch >= '0' && ch <= '9'){x = x * 10 + ch - '0';ch = getchar();} return x * f; } int n,head[MN+5],cnt=0,q,size[MN+5],mx[MN+5],dep[MN+5],nl[MN+5],nr[MN+5],dn=0,top[MN+5],fa[MN+5]; struct edge{int to,next;}e[MN+5]; inline void ins(int f,int t){e[++cnt]=(edge){t,head[f]};head[f]=cnt;} struct Tree{int l,r,x,val,tag;}T[MN*4+5]; void Pre(int x) { size[x]=1;mx[x]=0; for(int i=head[x];i;i=e[i].next) { dep[e[i].to]=dep[x]+1; Pre(e[i].to); size[x]+=size[e[i].to]; if(size[e[i].to]>size[mx[x]]) mx[x]=e[i].to; } } void Dfs(int x,int tp) { top[x]=tp;nl[x]=++dn; if(mx[x]) Dfs(mx[x],tp); for(int i=head[x];i;i=e[i].next) if(e[i].to!=mx[x]) Dfs(e[i].to,e[i].to); nr[x]=dn; } void Build(int x,int l,int r) { if((T[x].l=l)==(T[x].r=r)) return; int mid=l+r>>1; Build(x<<1,l,mid); Build(x<<1|1,mid+1,r); } inline void mark(int x,int v) { T[x].tag=1;T[x].val=v; T[x].x=(T[x].r-T[x].l+1)*v; } void pushdown(int x) { T[x].tag=0; mark(x<<1,T[x].val); mark(x<<1|1,T[x].val); } int Query(int x,int l,int r,int ad) { if(T[x].l==l&&T[x].r==r) { int ans=T[x].x;mark(x,ad); return ans; } if(T[x].tag) pushdown(x); int mid=T[x].l+T[x].r>>1,ans=0; if(r<=mid) ans=Query(x<<1,l,r,ad); else if(l>mid) ans=Query(x<<1|1,l,r,ad); else ans=Query(x<<1,l,mid,ad)+Query(x<<1|1,mid+1,r,ad); return T[x].x=T[x<<1].x+T[x<<1|1].x,ans; } int Solve_Chain(int x) { int sum=0; for(;x;x=fa[top[x]]) sum+=dep[x]-dep[top[x]]+1-Query(1,nl[top[x]],nl[x],1); return sum; } int Solve_Tree(int x){return Query(1,nl[x],nr[x],0);} char op[20]; int main() { n=read(); for(int i=2;i<=n;++i) fa[i]=read()+1,ins(fa[i],i); Pre(1);Dfs(1,1);Build(1,1,n);q=read(); for(int i=1;i<=q;++i) { scanf("%s",op+1);int x=read()+1; if(op[1]=='i') printf("%d ",Solve_Chain(x)); if(op[1]=='u') printf("%d ",Solve_Tree (x)); } return 0; }