【Bzoj4196】软件包管理器

题目描述

Linux用户和OSX用户一定对软件包管理器不会陌生。通过软件包管理器，你可以通过一行命令安装某一个软件包，然后软件包管理器会帮助你从软件源下载软件包，同时自动解决所有的依赖（即下载安装这个软件包的安装所依赖的其它软件包），完成所有的配置。ebian/Ubuntu使用的apt-get，Fedora/CentOS使用的yum，以及OSX下可用的homebrew都是优秀的软件包管理器。

你决定设计你自己的软件包管理器。不可避免地，你要解决软件包之间的依赖问题。如果软件包A依赖软件包B，那么安装软件包A以前，必须先安装软件包B。同时，如果想要卸载软件包B，则必须卸载软件包A。现在你已经获得了所有的软件包之间的依赖关系。而且，由于你之前的工作，除0号软件包以外，在你的管理器当中的软件包都会依赖一个且仅一个软件包，而0号软件包不依赖任何一个软件包。依赖关系不存在环（若有m(m≥2)个软件包A1,A2,A3,⋯,Am，其中A1依赖A2，A2依赖A3，A3依赖A4，……，A[m-1]依赖Am，而Am依赖A1，则称这m个软件包的依赖关系构成环），当然也不会有一个软件包依赖自己。

现在你要为你的软件包管理器写一个依赖解决程序。根据反馈，用户希望在安装和卸载某个软件包时，快速地知道这个操作实际上会改变多少个软件包的安装状态（即安装操作会安装多少个未安装的软件包，或卸载操作会卸载多少个已安装的软件包），你的任务就是实现这个部分。注意，安装一个已安装的软件包，或卸载一个未安装的软件包，都不会改变任何软件包的安装状态，即在此情况下，改变安装状态的软件包数为0。

-----------------------------------------------------------------------------------------------------------------------------

把题目审完之后发现就是两个操作，安装操作就是将软件包到根全都赋为1，然后输出改变的结点数量。删除操作就是将以软件包为根的子树全部赋为0，并输出修改数量。

#include <cstdio>
#include <algorithm>
#include <cstring>

using namespace std;

const int maxx = 100000 + 25;

int head[maxx],to[maxx<<1],nxt[maxx<<1],fnl[maxx];
int ftr[maxx],son[maxx],top[maxx],dpt[maxx],size[maxx],rnk[maxx];
int T[maxx<<2],Add[maxx<<2];

int n,m,num,cnt,x,y;
char flag[12];

void Ins(int x,int y) {to[++num]=y;nxt[num]=head[x];head[x]=num;}

namespace Y{
    
    void Dfs1(int x){
        size[x] = 1;
        for(int i=head[x];i;i=nxt[i]){
            int now=to[i];if(now == ftr[x]) continue;
            ftr[now] = x;dpt[now] = dpt[x] + 1;
            Dfs1(now);size[x] += size[now];
            if(size[now] > size[son[x]]) son[x] = now;
        }
    }
    
    void Dfs2(int x,int brn){
        rnk[x] = ++cnt;top[x] = brn;
        if(son[x]) Dfs2(son[x],brn);
        for(int i=head[x];i;i=nxt[i])
            if(to[i]!=ftr[x] && to[i]!=son[x])
                Dfs2(to[i],to[i]);
        fnl[x] = cnt;
    }
    
    void pushdown(int i,int l,int r){
        int mid = (l+r) >> 1;
        T[i<<1] = (mid-l+1)*Add[i];T[i<<1|1] = (r-mid)*Add[i];
        Add[i<<1] = Add[i<<1|1] = Add[i];Add[i] = -1;
    }
    
    void modify(int i,int x,int y,int l,int r,int k){
        if(x<=l && r<=y) {T[i]=(r-l+1)*k;Add[i]=k;return;}
        if(Add[i]!=-1) pushdown(i,l,r);int mid = (l+r) >> 1;
        if(x <= mid) modify(i<<1,x,y,l,mid,k);
        if(y >  mid) modify(i<<1|1,x,y,mid+1,r,k);
        T[i] = T[i<<1] + T[i<<1|1];
    }
    
    int Query(int i,int x,int y,int l,int r){
        if(x<=l && r<=y) return T[i];        
        int ans = 0;int mid = (l+r) >> 1;
        if(Add[i]!=-1) pushdown(i,l,r);
        if(x <= mid) ans += Query(i<<1,x,y,l,mid);
        if(y >  mid) ans += Query(i<<1|1,x,y,mid+1,r);
        return ans;
    }
    
    int solve(int x,int y){
        int ans = 0,tmp = dpt[y]+1;
        while(top[x] != top[y]){
            if(dpt[top[x]] > dpt[top[y]]) x^=y^=x^=y;
            ans += Query(1,rnk[top[y]],rnk[y],1,n);
            modify(1,rnk[top[y]],rnk[y],1,n,1);
            y = ftr[top[y]];
        }
        if(rnk[x] > rnk[y]) x^=y^=x^=y;
        ans += Query(1,rnk[x],rnk[y],1,n);
        modify(1,rnk[top[y]],rnk[y],1,n,1);
        return tmp-ans;
    }
    
}

using namespace Y;

int main(){
    memset(Add,-1,sizeof(Add));
    scanf("%d",&n);
    for(int i=2;i<=n;i++) scanf("%d",&x),Ins(i,x+1),Ins(x+1,i);
    Dfs1(1);Dfs2(1,1);
    scanf("%d",&m);
    while( m-- ){
        scanf("%s%d",flag,&x);x++;
        if(flag[0] == 'i') 
            printf("%d\n",solve(1,x));
        if(flag[0] == 'u'){
            printf("%d\n",Query(1,rnk[x],fnl[x],1,n));
            modify(1,rnk[x],fnl[x],1,n,0);
        }
    }
    return 0;
}