[Bzoj4196] [NOI2015] 软件包管理器 [树链剖分，线段树]

题解摘要：树链剖分后用线段树区间查询修改，对于安装软件，将改点到根的路径全部变为1，对于卸载软件，将子树清空。注意边界，编号是从0开始的，容易漏掉树根。

第一次写树剖～

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <cmath>
#include <ctime>

using namespace std;

struct Edge
{
    int    to,next;
}e[210000];

int    ind,n,Pre[210000],f[810000],Sum[810000],Out_dfn[210000];
int    p[210000],cnt,hv[210000],Head[210000],Dfn[210000];

void    Add_edge(const int & x,const int & y)
{
    e[++cnt].to=y;
    e[cnt].next=p[x];
    p[x]=cnt;
    return ;
}

int    Init_Dfs(const int& S)
{
    int    num=1,i,Max_=0;
    for(i=p[S];i;i=e[i].next)
    {
        int temp=Init_Dfs(e[i].to);
        if(temp>Max_)Max_=temp,hv[S]=e[i].to;
        num+=temp;
    }
    return num;
}

void    Dfs(const int& S,const int& fa)
{
    int    i;
    if(!Head[S])Head[S]=Head[fa];Dfn[S]=++ind;
    
    if(hv[S])Dfs(hv[S],S);
    for(i=p[S];i;i=e[i].next)
    {
        if(e[i].to!=hv[S])
        {
            Head[e[i].to]=e[i].to;
            Pre[e[i].to]=S;
            Dfs(e[i].to,S);
        }
    }
    Out_dfn[S]=ind;
    return ;
}

inline void    push_down(const int & num,const int & l,const int & r)
{
    if(f[num])
    {
        if(l!=r)
        {
            f[num<<1]=f[num<<1|1]=f[num];
        }
        if(f[num]==1)Sum[num]=r-l+1;
        if(f[num]==-1)Sum[num]=0;
        f[num]=0;
    }
    return ;
}

void    push_up(const int & num,const int & l,const int & r)
{
    if(l!=r)
    {
        int    mid=l+((r-l)>>1);
        push_down(num<<1,l,mid);
        push_down(num<<1|1,mid+1,r);
        Sum[num]=Sum[num<<1]+Sum[num<<1|1];
    }
    return ;
}

void    Change(const int & l,const int & r,const int & num,
        const int & s,const int & t,const int &d)
{
    if(s<=l && r<=t)
    {
        f[num]=d;
        push_down(num,l,r);
        return ;
    }
    int    mid=l+((r-l)>>1);
    push_down(num,l,r);
    if(s<=mid)Change(l,mid,num<<1,s,t,d);
    if(t>mid) Change(mid+1,r,num<<1|1,s,t,d);
    push_up(num,l,r);
    return ;
}

int    Query(const int & l,const int & r,const int & num,
        const int & s,const int & t)
{
    if(s<=l && r<=t)
    {
        push_down(num,l,r);
        return Sum[num];
    }

    int    mid=l+((r-l)>>1),temp=0;
    push_down(num,l,r);
    if(s<=mid)temp+=Query(l,mid,num<<1,s,t);
    if(t>mid) temp+=Query(mid+1,r,num<<1|1,s,t);
    push_up(num,l,r);
    return temp;
}

int    Get_sum_install(const int& S)
{
    int    x=S,temp=0,sum=0;
    
    while(x)
    {
        sum+=Query(1,n,1,Dfn[Head[x]],Dfn[x]);
        Change(1,n,1,Dfn[Head[x]],Dfn[x],1);
        temp+=Dfn[x]-Dfn[Head[x]]+1;
        x=Pre[Head[x]];
    }
    
    return temp-sum;
}

int    Get_sum_remove(const int & S)
{
    int    x=S,sum=0;

    sum=Query(1,n,1,Dfn[x],Out_dfn[x]);
    Change(1,n,1,Dfn[x],Out_dfn[x],-1);

    return sum;
}
    
int main()
{

    int    i,x,q;
    char    op[15];

    scanf("%d",&n);
    for(i=2;i<=n;++i)
    {
        scanf("%d",&x);x++;
        Add_edge(x,i);
    }

    Init_Dfs(1);

    Head[1]=1;
    Dfs(1,1);
    scanf("%d",&q);
    for(i=1;i<=q;++i)
    {
        scanf("%s%d",op,&x);x++;
        if(op[0]=='i')
            printf("%d
",Get_sum_install(x));
        else
            printf("%d
",Get_sum_remove(x));
    }

    return 0;
}

不知道为什么，BZOJ TLE了，但是UOJ AC（求大神指教）？？