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

题目链接

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

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

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

 #include<bits/stdc++.h>
 using namespace std;
 typedef long long ll;
 const int N=1e5+,inf=0x3f3f3f3f;
 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<<],mk[N<<];
 struct E {int v,nxt;} e[N];
 void addedge(int u,int v) {e[ne]= {v,hd[u]},hd[u]=ne++;}
 void dfs1(int u,int f,int d) {
     fa[u]=f,son[u]=,siz[u]=,dep[u]=d;
     for(int i=hd[u]; ~i; i=e[i].nxt) {
         int v=e[i].v;
         if(v==fa[u])continue;
         dfs1(v,u,d+),siz[u]+=siz[v];
         if(siz[v]>siz[son[u]])son[u]=v;
     }
 }
 void dfs2(int u,int tp) {
     top[u]=tp,dfn[u]=++tot,rnk[tot]=u;
     if(son[u])dfs2(son[u],tp);
     for(int i=hd[u]; ~i; i=e[i].nxt) {
         int v=e[i].v;
         if(v==fa[u]||v==son[u])continue;
         dfs2(v,v);
     }
     ed[u]=tot;
 }
 #define ls (u<<1)
 #define rs (u<<1|1)
 #define mid ((l+r)>>1)
 void pu(int u) {sum[u]=sum[ls]+sum[rs];}
 void pd(int u,int l,int r) {
     if(~mk[u]) {
         sum[u]=mk[u]*(r-l+);
         if(l<r)mk[ls]=mk[rs]=mk[u];
         mk[u]=-;
     }
 }
 void upd(int L,int R,int x,int u=,int l=,int r=tot) {
     pd(u,l,r);
     if(l>=L&&r<=R) {mk[u]=x,pd(u,l,r); return;}
     if(l>R||r<L)return;
     upd(L,R,x,ls,l,mid),upd(L,R,x,rs,mid+,r),pu(u);
 }
 int qry(int L,int R,int u=,int l=,int r=tot) {
     pd(u,l,r);
     if(l>=L&&r<=R)return sum[u];
     if(l>R||r<L)return ;
     return qry(L,R,ls,l,mid)+qry(L,R,rs,mid+,r);
 }
 void changeroot(int u,int x) {for(; u; u=fa[top[u]])upd(dfn[top[u]],dfn[u],x);}
 int askroot(int u) {
     int ret=;
     for(; u; u=fa[top[u]])ret+=qry(dfn[top[u]],dfn[u]);
     return ret;
 }
 void changesub(int u,int x) {upd(dfn[u],ed[u],x);}
 int asksub(int u) {return qry(dfn[u],ed[u]);}
 int main() {
     memset(hd,-,sizeof hd),ne=;
     scanf("%d",&n);
     for(int i=; i<=n; ++i) {
         int u;
         scanf("%d",&u),u++;
         addedge(u,i);
     }
     tot=,dfs1(,,),dfs2(,);
     memset(mk,-,sizeof mk);
     char s[];
     scanf("%d",&m);
     while(m--) {
         int u;
         scanf("%s%d",s,&u),u++;
         if(s[]=='i') {
             printf("%d\n",dep[u]-askroot(u));
             changeroot(u,);
         } else {
             printf("%d\n",asksub(u));
             changesub(u,);
         }
     }
     return ;
 }