BZOJ_P4196 [NOI2015]软件包管理器(树链剖分+dfs序)

BZOJ传送门

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

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

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

输入格式
输入文件的第 11 行包含 11 个正整数 nn，表示软件包的总数。软件包从 00 开始编号。

随后一行包含 n−1n−1 个整数，相邻整数之间用单个空格隔开，分别表示 1,2,3,…,n−2,n−11,2,3,…,n−2,n−1 号软件包依赖的软件包的编号。

接下来一行包含 11 个正整数 qq，表示询问的总数。

之后 qq 行，每行 11 个询问。询问分为两种：

install xx：表示安装软件包 xx
uninstall xx：表示卸载软件包 xx
你需要维护每个软件包的安装状态，一开始所有的软件包都处于未安装状态。对于每个操作，你需要输出这步操作会改变多少个软件包的安装状态，随后应用这个操作（即改变你维护的安装状态）。

输出格式
输出文件包括 qq 行。

输出文件的第 ii 行输出 11 个整数，为第 ii 步操作中改变安装状态的软件包数。

样例一
input

7
0 0 0 1 1 5
5
install 5
install 6
uninstall 1
install 4
uninstall 0

output

3
1
3
2
3

explanation

一开始所有的软件包都处于未安装状态。

安装 55 号软件包，需要安装 0,1,50,1,5 三个软件包。

之后安装 66 号软件包，只需要安装 66 号软件包。此时安装了 0,1,5,60,1,5,6 四个软件包。

卸载 11 号软件包需要卸载 1,5,61,5,6 三个软件包。此时只有 00 号软件包还处于安装状态。

之后安装 44 号软件包，需要安装 1,41,4 两个软件包。此时 0,1,40,1,4 处在安装状态。

最后，卸载 00 号软件包会卸载所有的软件包。

样例二
input

10
0 1 2 1 3 0 0 3 2
10
install 0
install 3
uninstall 2
install 7
install 5
install 9
uninstall 9
install 4
install 1
install 9

output

1
3
2
1
3
1
1
1
0
1

样例三
见样例数据下载。

限制与约定
测试点编号 nn 的规模 qq 的规模 备注
1 n=5000n=5000 q=5000q=5000
2
3 n=100000n=100000 q=100000q=100000 数据不包含卸载操作
4
5 n=100000n=100000 q=100000q=100000 编号为 ii 的软件包所依赖的软件包编号在 [0,i−1][0,i−1] 内均匀随机。
每次执行操作的软件包编号在 [0,n−1][0,n−1] 内均匀随机。
6
7
8
9 n=100000n=100000 q=100000q=100000
10
11
12
13
14
15
16
17
18
19
20
时间限制：1s1s
空间限制：512MB

Sol:
NOI水题…

#include<cstdio>
#include<vector>
#include<iostream>
using namespace std;
#define N 500005
#define mid ((l+r)>>1)
int n,m,cnt;vector<int> g[N];
int sz[N],top[N],p[N],f[N],son[N],dep[N],dfsn[N];
struct SegTree{
    int d[N*4],g[N*4],lg[N*4];
    void Updata(int o){d[o]=d[o<<1]+d[o<<1|1];}
    void Push(int o){
        if(~g[o]){
            if(g[o]) d[o<<1]=lg[o<<1],d[o<<1|1]=lg[o<<1|1];
            else d[o<<1]=d[o<<1|1]=0;
            g[o<<1]=g[o<<1|1]=g[o],g[o]=-1;
        }
    }
    void Build(int o,int l,int r){
        if(l==r){d[o]=0,g[o]=-1,lg[o]=1;return;}
        Build(o<<1,l,mid);Build(o<<1|1,mid+1,r);
        lg[o]=lg[o<<1]+lg[o<<1|1],g[o]=-1;
    }
    int Query0(int o,int l,int r,int ql,int qr,int res=0){
        if(ql<=l&&r<=qr) return lg[o]-d[o];
        Push(o);
        if(ql<=mid) res+=Query0(o<<1,l,mid,ql,qr);
        if(qr>mid) res+=Query0(o<<1|1,mid+1,r,ql,qr);
        return res;
    }
    int Query1(int o,int l,int r,int ql,int qr,int res=0){
        if(ql<=l&&r<=qr) return d[o];
        Push(o);
        if(ql<=mid) res+=Query1(o<<1,l,mid,ql,qr);
        if(qr>mid) res+=Query1(o<<1|1,mid+1,r,ql,qr);
        return res;
    }
    void Change(int o,int l,int r,int ql,int qr,int v){
        if(ql<=l&&r<=qr){g[o]=v,d[o]=lg[o]*v;return;}
        Push(o);
        if(ql<=mid) Change(o<<1,l,mid,ql,qr,v);
        if(qr>mid) Change(o<<1|1,mid+1,r,ql,qr,v);
        Updata(o);
    }
    int GetAns(int u,int v,int res=0){
        int f1=top[u],f2=top[v];
        while(f1!=f2){
            if(dep[f1]<dep[f2]) swap(u,v),swap(f1,f2);
            res+=Query0(1,1,n,p[f1],p[u]);Change(1,1,n,p[f1],p[u],1);
            u=f[f1],f1=top[u];
        }
        if(dep[u]>dep[v]) swap(u,v);
        res+=Query0(1,1,n,p[u],p[v]);Change(1,1,n,p[u],p[v],1);
        return res;
    }
}seg;
void dfs1(int u,int fa,int d){
    f[u]=fa,sz[u]=1,dep[u]=d;
    for(int i=0,v,lim=g[u].size();i<lim;i++)
        if((v=g[u][i])!=fa){
            dfs1(v,u,d+1),sz[u]+=sz[v];
            if(!son[u]||sz[v]>sz[son[u]]) son[u]=v;
        }
}
void dfs2(int u,int tp){
    p[u]=++cnt,top[u]=tp;
    if(!son[u]){dfsn[u]=cnt;return;}dfs2(son[u],tp);
    for(int i=0,v,lim=g[u].size();i<lim;i++){
        if((v=g[u][i])!=f[u]&&v!=son[u]) dfs2(v,v);
    }dfsn[u]=cnt;
}
inline int in(int x=0,char ch=getchar()){while(ch>'9'||ch<'0') ch=getchar();
    while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+ch-'0',ch=getchar();return x;}
char str[65];
void print(long long x){
    int l=0;if(!x){putchar('0');return;}if(x<0) putchar('-'),x=-x;
    while(x) str[++l]=x%10+'0',x/=10;while(l) putchar(str[l--]);}
int main(){
    n=in();
    for(int i=2,x;i<=n;i++) x=in()+1,g[x].push_back(i);
    dfs1(1,1,1);dfs2(1,1);seg.Build(1,1,n);
    m=in();char opt[15];
    while(m--){
        scanf("%s",opt);
        if(opt[0]=='i'){
            print(seg.GetAns(1,in()+1)),putchar('\n');
        }else{
            int x=in()+1;
            print(seg.Query1(1,1,n,p[x],dfsn[x])),putchar('\n'),
                seg.Change(1,1,n,p[x],dfsn[x],0);
        }
    }
    return 0;
}