BZOJ - 4196 软件包管理器 (树链剖分+dfs序+线段树)

设白色结点为未安装的软件，黑色结点为已安装的软件，则：

安装软件i：输出结点i到根的路径上的白色结点的数量，并把结点i到根的路径染成黑色。复杂度$O(nlog^2n)$

卸载软件i：输出结点i子树下的黑色结点数量，并把结点i的子树染成白色。复杂度$O(nlogn)$

 1 #include<bits/stdc++.h>
 2 using namespace std;
 3 typedef long long ll;
 4 const int N=1e5+10,inf=0x3f3f3f3f;
 5 int n,m,hd[N],ne,fa[N],son[N],siz[N],dep[N],top[N],dfn[N],rnk[N],ed[N],tot,sum[N<<2],mk[N<<2];
 6 struct E {int v,nxt;} e[N];
 7 void addedge(int u,int v) {e[ne]= {v,hd[u]},hd[u]=ne++;}
 8 void dfs1(int u,int f,int d) {
 9     fa[u]=f,son[u]=0,siz[u]=1,dep[u]=d;
10     for(int i=hd[u]; ~i; i=e[i].nxt) {
11         int v=e[i].v;
12         if(v==fa[u])continue;
13         dfs1(v,u,d+1),siz[u]+=siz[v];
14         if(siz[v]>siz[son[u]])son[u]=v;
15     }
16 }
17 void dfs2(int u,int tp) {
18     top[u]=tp,dfn[u]=++tot,rnk[tot]=u;
19     if(son[u])dfs2(son[u],tp);
20     for(int i=hd[u]; ~i; i=e[i].nxt) {
21         int v=e[i].v;
22         if(v==fa[u]||v==son[u])continue;
23         dfs2(v,v);
24     }
25     ed[u]=tot;
26 }
27 #define ls (u<<1)
28 #define rs (u<<1|1)
29 #define mid ((l+r)>>1)
30 void pu(int u) {sum[u]=sum[ls]+sum[rs];}
31 void pd(int u,int l,int r) {
32     if(~mk[u]) {
33         sum[u]=mk[u]*(r-l+1);
34         if(l<r)mk[ls]=mk[rs]=mk[u];
35         mk[u]=-1;
36     }
37 }
38 void upd(int L,int R,int x,int u=1,int l=1,int r=tot) {
39     pd(u,l,r);
40     if(l>=L&&r<=R) {mk[u]=x,pd(u,l,r); return;}
41     if(l>R||r<L)return;
42     upd(L,R,x,ls,l,mid),upd(L,R,x,rs,mid+1,r),pu(u);
43 }
44 int qry(int L,int R,int u=1,int l=1,int r=tot) {
45     pd(u,l,r);
46     if(l>=L&&r<=R)return sum[u];
47     if(l>R||r<L)return 0;
48     return qry(L,R,ls,l,mid)+qry(L,R,rs,mid+1,r);
49 }
50 void changeroot(int u,int x) {for(; u; u=fa[top[u]])upd(dfn[top[u]],dfn[u],x);}
51 int askroot(int u) {
52     int ret=0;
53     for(; u; u=fa[top[u]])ret+=qry(dfn[top[u]],dfn[u]);
54     return ret;
55 }
56 void changesub(int u,int x) {upd(dfn[u],ed[u],x);}
57 int asksub(int u) {return qry(dfn[u],ed[u]);}
58 int main() {
59     memset(hd,-1,sizeof hd),ne=0;
60     scanf("%d",&n);
61     for(int i=2; i<=n; ++i) {
62         int u;
63         scanf("%d",&u),u++;
64         addedge(u,i);
65     }
66     tot=0,dfs1(1,0,1),dfs2(1,1);
67     memset(mk,-1,sizeof mk);
68     char s[10];
69     scanf("%d",&m);
70     while(m--) {
71         int u;
72         scanf("%s%d",s,&u),u++;
73         if(s[0]=='i') {
74             printf("%d\n",dep[u]-askroot(u));
75             changeroot(u,1);
76         } else {
77             printf("%d\n",asksub(u));
78             changesub(u,0);
79         }
80     }
81     return 0;
82 }