NOI2015 软件包管理器 manager

显然链剖 然而只询问到根的信息，不用管lca，要好些很多（虽然我没那么写）

对于安装 查询和维护到根路径

对于卸载 查询和维护子树信息

因为链剖本身是用dfs序建的线段树，所以使得查询和修改子树非常方便。 

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

using namespace std;

const int Maxn=100010;
int n;
struct Data{
    int num[2];
    Data(int a=0,int b=0) {num[0]=a,num[1]=b;}
    Data operator + (const Data&rhs) const {
        return Data(num[0]+rhs.num[0],num[1]+rhs.num[1]);
    }
};

struct  SegmentTree{
    Data da[Maxn*4];
    int tag[Maxn*4];
    int lft,rgt,w;
    
    void add_tag(int s,int l,int r,int w) {
        if(w==-1) return;
        da[s].num[w]=r-l+1;
        da[s].num[w^1]=0;
        tag[s]=w;
    }
    
    void down(int s,int l,int r) {
        if(tag[s]==-1 || l==r) return;
        int mid=(l+r)>>1;
        add_tag(s<<1,l,mid,tag[s]);
        add_tag(s<<1|1,mid+1,r,tag[s]);
        tag[s]=-1;
    }
    
    void change(int s,int l,int r) {
        if(lft<=l && r<=rgt) {
            add_tag(s,l,r,w);
            return;
        }
        down(s,l,r);
        int mid=(l+r)>>1;
        if(lft<=mid) change(s<<1,l,mid);
        if(mid<rgt) change(s<<1|1,mid+1,r);
        da[s] = da[s<<1] + da[s<<1|1];
    }
    Data query(int s,int l,int r) {
        if(lft<=l&&r<=rgt) return da[s];
        down(s,l,r);
        int mid=(l+r)>>1;
        if(rgt<=mid) return query(s<<1,l,mid);
        if(mid<lft) return query(s<<1|1,mid+1,r);
        return query(s<<1,l,mid) + query(s<<1|1,mid+1,r);
    }
    void build(int s,int l,int r) {
        if(l==r) {
            da[s]=Data(1,0);
            return;
        }
        int mid=(l+r)>>1;
        build(s<<1,l,mid);
        build(s<<1|1,mid+1,r);
        da[s]=da[s<<1]+da[s<<1|1];
    }
}seg;

int sz[Maxn],fa[Maxn],son[Maxn],dep[Maxn];

struct Edge {
    int to;
    Edge*next;
    Edge(int to=0,Edge*next=0):to(to),next(next) {}
}pool[Maxn*2],*pis=pool,*fir[Maxn];

void AddEdge(int from,int to) {
    *pis=Edge(to,fir[from]);fir[from]=pis++;
    *pis=Edge(from,fir[to]);fir[to]=pis++;
}

void dfs(int u) {
    sz[u] = 1;
    son[u] = 0;
    for(Edge*p=fir[u];p;p=p->next) {
        int v=p->to;
        if(v==fa[u]) continue;
        fa[v]=u;
        dep[v]=dep[u]+1;
        dfs(v);
        sz[u]+=sz[v];
        if(sz[v]>sz[son[u]]) son[u]=v;
    }
}

int pos[Maxn],top[Maxn],tm,end[Maxn];

void build(int u,int pre) {
    top[u] = pre;
    pos[u] = ++tm;
    if(son[u]) build(son[u],pre);
    for(Edge*p=fir[u];p;p=p->next) {
        int v=p->to;
        if(v==fa[u] || v==son[u]) continue;
        build(v,v);
    }
    end[u]=tm;
}

Data query(int a,int b) {
    int ta=top[a],tb=top[b];
    Data ret;
    for(;ta!=tb;a=fa[ta],ta=top[a]) {
        if(dep[ta]<dep[tb]) swap(a,b) ,swap(ta,tb);
        seg.lft=pos[ta],seg.rgt=pos[a];
        ret=ret+seg.query(1,1,n);
    }
    if(dep[a]>dep[b]) swap(a,b);
    seg.lft=pos[a],seg.rgt=pos[b];
    Data tmp=seg.query(1,1,n);
    return ret+tmp;
}

void change(int a,int b,int c) {
    int ta=top[a],tb=top[b];
    Data ret;
    for(;ta!=tb;a=fa[ta],ta=top[a]) {
        if(dep[ta]<dep[tb]) swap(a,b) ,swap(ta,tb);
        seg.lft=pos[ta],seg.rgt=pos[a];seg.w=c;
        seg.change(1,1,n);
    }
    if(dep[a]>dep[b]) swap(a,b);
    seg.lft=pos[a],seg.rgt=pos[b],seg.w=c;
    return seg.change(1,1,n);
}

void init() {
    memset(seg.tag,-1,sizeof seg.tag);
    scanf("%d",&n);
    for(int x,i=1;i<n;i++) {
        scanf("%d",&x);
        AddEdge(i+1,x+1);
    }
    dfs(1);
    build(1,1);
    seg.build(1,1,n);
    
//    seg.lft=1,seg.rgt=n,seg.w=0;change(1,1,n);
}

void work() {
    int m,x;
    char opt[100];
    for(scanf("%d",&m);m--;) {
        scanf("%s%d",opt,&x);x++;
        if(opt[0]=='i') {
            printf("%d
",query(x,1).num[0]);
            change(x,1,1);
        }else {
            seg.lft=pos[x],seg.rgt=end[x],seg.w=0;
            printf("%d
",seg.query(1,1,n).num[1]);
            seg.change(1,1,n);
        }
    }
}

int main() {
    freopen("manager.in","r",stdin);
    freopen("manager.out","w",stdout);
    
    init();
    work();
    
    return 0;
}