【NOI2015】软件包管理器（树链剖分）

https://loj.ac/problem/2130

显然这道题是一个树状结构。
对于两种操作
设未安装为0，已安装为1
install x，相当于查询从0到x的路径中有多少个0，再将路径中的所有值置为1
uninstall x，相当于查询x的子树中有多少个1，再将子树中的所有值置为0

其实就是模板题，为了方便实现，我将0和1互换，这样的话install过程直接区间求和与区间修改。
而uninstall过程子树求和后再用子树的大小减去求得和，得到子树中0的个数。修改直接修改子树即可。

#include<bits/stdc++.h>
#define fer(i,j,n) for(int i=j;i<=n;i++)
#define far(i,j,n) for(int i=j;i>=n;i--)
#define ll long long
#define pa pair<int,int>
const int maxn=1000010;
const int INF=1e9+7;
using namespace std;
/*----------------------------------------------------------------------------*/
inline ll read()
{
    char ls;ll x=0,sng=1;
    for(;ls<'0'||ls>'9';ls=getchar())if(ls=='-')sng=-1;
    for(;ls>='0'&&ls<='9';ls=getchar())x=x*10+ls-'0';
    return x*sng;
}
/*----------------------------------------------------------------------------*/
ll n,m; 
struct kaga
{
    int next;
    int point;
}e[maxn];
struct akagi
{
    int l,r;
    ll sum,lazy;
}tr[maxn];
int last[maxn],k=0,fa[maxn],size[maxn],son[maxn],belong[maxn],hashh[maxn],hashv[maxn],dep[maxn],v[maxn];
void sumup(int x)
{
    tr[x].sum=tr[x<<1].sum+tr[x<<1|1].sum;
    return ;
}
void pushdown(int x)
{
    int y=tr[x].lazy;
    if(y==1)
    {
        tr[x<<1].lazy=1;
        tr[x<<1].sum=(tr[x<<1].r-tr[x<<1].l+1);
        tr[x<<1|1].lazy=1;
        tr[x<<1|1].sum=(tr[x<<1|1].r-tr[x<<1|1].l+1);   
    }
    if(y==2)
    {
        tr[x<<1].lazy=2;
        tr[x<<1].sum=0;
        tr[x<<1|1].lazy=2;
        tr[x<<1|1].sum=0;   
    }
    tr[x].lazy=0;
}
void build(int x,int l,int r)
{
    int mid=(l+r)>>1;
    tr[x].l=l;tr[x].r=r;
    if(l==r)
    {
        tr[x].sum=1;
        tr[x].lazy=0;
        return ;
    }
    build(x<<1,l,mid);
    build(x<<1|1,mid+1,r);
    sumup(x);
    return ;
}
void change(int x,int L,int R,int val)
{
    int l=tr[x].l,r=tr[x].r,mid=(l+r)>>1;
    if(L<=l&&r<=R)
    {
        if(val==1)
        {
            tr[x].sum=r-l+1;
            tr[x].lazy=1;   
        }
        else
        if(val==2)
        {
            tr[x].sum=0;
            tr[x].lazy=2;   
        }
        return ;
    }
    pushdown(x);
    if(R<=mid)change(x<<1,L,R,val);
    else if(L>mid)change(x<<1|1,L,R,val);
    else change(x<<1,L,mid,val),change(x<<1|1,mid+1,R,val);
    sumup(x);
    return ;
}
ll query(int x,int L,int R)
{
    int l=tr[x].l,r=tr[x].r,mid=(l+r)>>1;
    if(L<=l&&r<=R)return tr[x].sum;
    pushdown(x);
    if(R<=mid)query(x<<1,L,R);
    else if(L>mid)query(x<<1|1,L,R);
    else return query(x<<1,L,mid)+query(x<<1|1,mid+1,R);
}
void add_edge(int x,int y)
{
    e[++k].next=y;e[k].point=last[x];last[x]=k;
}
void insert(int x,int y)
{
    add_edge(x,y);
    add_edge(y,x);
}
void dfs1(int x,int fat)
{
    fa[x]=fat;
    dep[x]=dep[fat]+1;
    size[x]=1;
    for(int j=last[x];j;j=e[j].point)
    {
        int y=e[j].next;
        if(y==fat)continue;
        dfs1(y,x);
        size[x]+=size[y];
        if(!son[x]||size[y]>size[son[x]])
        son[x]=y;   
    }
}
void dfs2(int x,int top)
{
    belong[x]=top;
    hashh[x]=++k;
    hashv[k]=x;
    if(!son[x])return ;
    dfs2(son[x],top);
    for(int j=last[x];j;j=e[j].point)
    {
        int y=e[j].next;
        if(y!=son[x]&&y!=fa[x])
        dfs2(y,y);  
    }
}
void update(int x,int y,int val)
{
    int topx=belong[x];
    int topy=belong[y];
    while(topx!=topy)
    {
        if(dep[topx]<dep[topy])
        {
            swap(x,y);
            swap(topx,topy);    
        }
        change(1,hashh[topx],hashh[x],val);
        x=fa[topx];
        topx=belong[x];
    }
    if(dep[x]>dep[y])swap(x,y);
    change(1,hashh[x],hashh[y],val);
}
ll sigma(int x,int y)
{
    ll sum=0;
    int topx=belong[x];
    int topy=belong[y];
    while(topx!=topy)
    {
        if(dep[topx]<dep[topy])
        {
            swap(x,y);
            swap(topx,topy);    
        }
        sum+=query(1,hashh[topx],hashh[x]);
        x=fa[topx];
        topx=belong[x];
    }
    if(dep[x]>dep[y])swap(x,y);
    sum+=query(1,hashh[x],hashh[y]);
    return sum;
}
void updateson(int x,int val)
{
    int left=hashh[x];
    int right=hashh[x]+size[x]-1;
    change(1,left,right,val);
}
ll sigmason(int x)
{
    int left=hashh[x];
    int right=hashh[x]+size[x]-1;
    return right-left+1-query(1,left,right);
}
int main()
{
    n=read();
    fer(i,2,n)
    {
        int x=read()+1;
        insert(x,i);
    }
    k=0;
    dfs1(1,0);
    dfs2(1,1);
    build(1,1,n);
    m=read();
    fer(i,1,m)
    {
        char ls[10];
        scanf("%s",ls);
        if(ls[0]=='i')
        {
            int x=read()+1;
            cout<<sigma(1,x)<<endl;
            update(1,x,2);
        }
        else
        {
            int x=read()+1;
            cout<<sigmason(x)<<endl;
            updateson(x,1);
        }
    }
}