bzoj4196: [Noi2015]软件包管理器

bzoj4196: [Noi2015]软件包管理器
一眼链剖，然而不知道为什么写得很长/写得很差

---

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#define mxn 100010
using namespace std;
int n,q,u,s,t=1;
int dep[mxn],f[mxn],siz[mxn],son[mxn],
    pos[mxn],npos[mxn],top[mxn],a[mxn];
int to[mxn<<1],hd[mxn<<1],lk[mxn],cnt=0;
char ch[10];
void add(int st,int ed)
{to[++cnt]=ed,hd[cnt]=lk[st],lk[st]=cnt;}
void dfs(int k)
{
    siz[k]=1;
    dep[k]=dep[f[k]]+1;
    for(int i=lk[k];i;i=hd[i])
    if(to[i]!=f[k])
    {
        f[to[i]]=k;
        dfs(to[i]);
        siz[k]+=siz[to[i]];
        if(siz[to[i]]>siz[son[k]])
        son[k]=to[i];
    }
}
void dfss(int k)
{
    pos[t]=k,npos[k]=t++;
    if(son[k])
    top[son[k]]=top[k],dfss(son[k]);
    for(int i=lk[k];i;i=hd[i])
    if(to[i]!=son[k]&&to[i]!=f[k])
    top[to[i]]=to[i],dfss(to[i]);
}
int sum[mxn<<2],laz[mxn<<2],ans;
int ll[mxn<<2],rr[mxn<<2];
void update(int k)
{sum[k]=sum[k<<1]+sum[k<<1|1];}
void build(int k,int l,int r)
{
    ll[k]=l,rr[k]=r;
    if(l!=r)
    build(k<<1,l,(l+r)>>1),
    build(k<<1|1,((l+r)>>1)+1,r);
}
void pushdown(int k)
{
    laz[k<<1]=laz[k<<1|1]=laz[k];
    sum[k<<1]=(laz[k]-1)*(rr[k<<1]-ll[k<<1]+1);
    sum[k<<1|1]=(laz[k]-1)*(rr[k<<1|1]-ll[k<<1|1]+1);
    laz[k]=0;
}
int qm(int k,int l,int r,bool mode)
{
    int tmp; 
    if(ll[k]==l&&rr[k]==r)
    {
        tmp=sum[k];
        laz[k]=mode+1,sum[k]=mode*(r-l+1);
        return mode?r-l-tmp+1:tmp;
    }
    else
    {
        if(laz[k])pushdown(k);
        int mid=(ll[k]+rr[k])>>1;
        if(l>mid)tmp=qm(k<<1|1,l,r,mode);
        else if(r<=mid)tmp=qm(k<<1,l,r,mode);
        else tmp=qm(k<<1,l,mid,mode)+qm(k<<1|1,mid+1,r,mode);
        update(k);
        return tmp;
    }
}
int main()
{
    scanf("%d",&n);
    for(int i=2;i<=n;i++)
    {
        scanf("%d",&u);
        add(i,u+1),add(u+1,i);
    }
    dfs(1),dfss(1);
    build(1,1,n);
    scanf("%d",&q);
    while(q--)
    {
        scanf("%s%d",ch,&u);
        u++,ans=0;
        if(ch[0]=='i')
        {
            while(top[u]>1)
            ans+=qm(1,npos[top[u]],npos[u],1),u=f[top[u]];
            ans+=qm(1,1,npos[u],1);
        }
        else
        ans+=qm(1,npos[u],npos[u]+siz[u]-1,0);
        printf("%d\n",ans);
    }
}