bzoj4196 [Noi2015]软件包管理器——树链剖分

题目：https://www.lydsy.com/JudgeOnline/problem.php?id=4196

树链剖分。

代码如下：

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
int const maxn=1e5+;
int n,m,fa[maxn],dfn[maxn],end[maxn],top[maxn],to[maxn],siz[maxn],head[maxn],ct,tim;
struct N{
    int to,next;
    N(int t=,int n=):to(t),next(n) {}
}edge[maxn];
struct T{int sum,siz; bool f[];}t[maxn<<];
int rd()
{
    int ret=;char ch=getchar();
    while(ch<''||ch>'')ch=getchar();
    while(ch>=''&&ch<='')ret=ret*+ch-'',ch=getchar();
    return ret;
}
void dfs(int x)
{
    siz[x]=;
    for(int i=head[x],u;i;i=edge[i].next)
    {
        dfs(u=edge[i].to);
        if(siz[u]>siz[to[x]])to[x]=u;
        siz[x]+=siz[u];
    }
}
void dfs2(int x)
{
    dfn[x]=++tim;//不在dfs
    if(to[x])top[to[x]]=top[x],dfs2(to[x]);
    for(int i=head[x],u;i;i=edge[i].next)
    {
        if((u=edge[i].to)==to[x])continue;
        top[u]=u; dfs2(u);
    }
    end[x]=tim;
}
void pushup(int x)
{
    int ls=(x<<),rs=(x<<|);
    t[x].siz=t[ls].siz+t[rs].siz;
    t[x].sum=t[ls].sum+t[rs].sum;
}
void pushdown(int x)
{
    if(!t[x].f[]&&!t[x].f[])return;
    int ls=(x<<),rs=(x<<|);
    for(int i=;i<=;i++)
        if(t[x].f[i])
        {
            t[x].f[i]=; t[ls].f[i]=; t[rs].f[i]=;
            t[ls].f[!i]=; t[rs].f[!i]=;
            t[ls].sum=t[ls].siz*i; t[rs].sum=t[rs].siz*i;
        }
}
void build(int x,int l,int r)
{
    if(l==r){t[x].siz=;return;}
    int mid=((l+r)>>);
    build(x<<,l,mid); build(x<<|,mid+,r);
    pushup(x);
}
int query(int x,int l,int r,int L,int R,int val)
{
    if(l>=L&&r<=R)
    {
        if(val)return t[x].sum;
        else return t[x].siz-t[x].sum;
    }
    int mid=((l+r)>>),ret=;
    pushdown(x);
    if(mid>=L)ret+=query(x<<,l,mid,L,R,val);
    if(mid<R)ret+=query(x<<|,mid+,r,L,R,val);
    return ret;
}
void update(int x,int l,int r,int L,int R,int val)
{
    if(l>=L&&r<=R)
    {
        t[x].f[val]=;  t[x].f[!val]=;
        t[x].sum=t[x].siz*val; return;
    }
    int mid=((l+r)>>);
    if(mid>=L)update(x<<,l,mid,L,R,val);
    if(mid<R)update(x<<|,mid+,r,L,R,val);
    pushup(x);
}
int ask(int x)
{
    int ret=;
    while(x)
    {
        int l=dfn[top[x]],r=dfn[x];
        ret+=query(,,n,l,r,);
        update(,,n,l,r,);
        x=fa[top[x]];
    }
    return ret;
}
int main()
{
    n=rd();
    for(int i=;i<=n;i++)
    {
        fa[i]=rd()+;
        edge[++ct]=N(i,head[fa[i]]);head[fa[i]]=ct;
    }
    dfs(); top[]=; dfs2(); build(,,n);
    m=rd(); char ch[];
    for(int i=,x;i<=m;i++)
    {
        scanf("%s",&ch); x=rd()+;
        if(ch[]=='u')
        {
            printf("%d\n",query(,,n,dfn[x],end[x],));
            update(,,n,dfn[x],end[x],);
        }
        if(ch[]=='i')
        {
            printf("%d\n",ask(x));
        }
    }
    return ;
}