【BZOJ4196】【Noi2015】软件包管理器

原题传送门

题意：

给你一棵树，有2种操作：

　　1.使得某个点到根节点路径上的所有点权值赋为1。

　　2.使得某节点的子树中所有节点权值赋为0.

每次操作要求输出权值更改的节点个数。

解题思路：

　　显然是用树剖来解决问题，考虑用区间赋值的线段树维护区间内1的个数，然后直接按题意树剖求解即可。

　　　　操作1的时间效率为\( O ( \log^{2} n) \) ,操作2的时间效率为\( O( \log n) \) ,总时间复杂度最坏为\( O (m \log^{2} n) \).

　　　　BZOJ上跑了5900ms，6960KB

#include <stdio.h>
#define MN 100005
#define Mn (1<<17)
#define ls (k<<1)
#define rs (k<<1|1)
#define mid (l+r>>1)
#define v edge[i].to
int mark[Mn<<],sum[Mn<<],siz[MN],dep[MN],son[MN],fa[MN],top[MN],head[MN],cnt,n,q,dfsn,l[MN],r[MN];
struct zxy{int to,nxt;}edge[MN];
inline void ins(int x,int y){edge[++cnt].to=y,edge[cnt].nxt=head[x],head[x]=cnt;}
inline int in(){
    int x=,f=;char ch=getchar();
    while (ch<''||ch>'') f=ch=='-'?-:,ch=getchar();
    while (ch>=''&&ch<='') x=x*+ch-'',ch=getchar();
    return x*f;
}
inline void dfs1(int u,int d){
    dep[u]=d,siz[u]=,son[u]=-;
    for (register int i=head[u]; i; i=edge[i].nxt){
        dfs1(v,d+);siz[u]+=siz[v];
        if (son[u]==-||siz[v]>siz[son[u]]) son[u]=v;
    }
}
inline void dfs2(int u,int tp){
    top[u]=tp;l[u]=(++dfsn);if (son[u]!=-) dfs2(son[u],tp);
    for (register int i=head[u]; i; i=edge[i].nxt)
        if (v!=son[u]) dfs2(v,v);r[u]=dfsn;
}
inline void pushdown(int k,int l,int r){
    if (mark[k]==-)return;
    register int x=r-l+;mark[ls]=mark[rs]=mark[k];
    sum[ls]=mark[k]*(x-(x>>));sum[rs]=mark[k]*(x>>);mark[k]=-;
}
inline void A(int l,int r,int a,int b,int k,int ad){
    if (a<=l&&r<=b) {mark[k]=ad; sum[k]=(r-l+)*ad;return;}
    pushdown(k,l,r);if (a<=mid) A(l,mid,a,b,ls,ad);
    if (b>mid) A(mid+,r,a,b,rs,ad);sum[k]=sum[ls]+sum[rs];
}
inline int Q(int l,int r,int a,int b,int k){
    if (a==l&&b==r) return sum[k];pushdown(k,l,r);
    if (b<=mid) return Q(l,mid,a,b,ls);
    if (mid<a) return Q(mid+,r,a,b,rs);
    return Q(l,mid,a,mid,ls)+Q (mid+,r,mid+,b,rs);
}
inline int install(int x){
    register int res=;
    while (top[x]) res+=l[x]-l[top[x]]-Q(,n,l[top[x]],l[x],)+,A(,n,l[top[x]],l[x],,),x=fa[top[x]];
    res+=l[x]-l[]+-Q(,n,l[],l[x],);A(,n,l[],l[x],,);return res;
}
inline int uninstall(int x){
    register int res=Q(,n,l[x],r[x],);
    A(,n,l[x],r[x],,);return res;
}
void init(){
    n=in();for (register int i=; i<n; ++i) fa[i]=in(),ins(fa[i],i);
    dfs1(,);dfs2(,);
}
void solve(){
    for (q=in(); q--;){
        register char op[];scanf("%s",op);register int x=in();
        if (op[]=='i') printf("%d\n",install(x));
        else printf("%d\n",uninstall(x));
    }
}
int main(){init();solve();return ;}