P2146 [NOI2015]软件包管理器 ( 树链剖分

题意:

树剖水题:

操作1. 将一个点到根节点之间的序列, 统计
操作2.将一颗子树的内部节点, 统计

//
// Created by Yishui on 2018/9/15.
//
#include <bits/stdc++.h>
using namespace std;

#define cpp_io() {ios::sync_with_stdio(false); cin.tie(NULL);}
#define rep(i,a,n) for (int i=a;i<n;i++)
#define repp(i,a,n) for (int i=a;i<=n;i++)
#define per(i,a,n) for (int i=n-1;i>=a;i--)
#define CLR(a,b) memset(a,(b),sizeof(a))
#define all(x) (x).begin(),(x).end()
#define SZ(x) ((int)(x).size())
#define pb push_back
#define mp make_pair
#define fi first
#define se second
#define ls o<<1
#define rs o<<1|1


typedef long long ll;
typedef vector<int> VI;
const int MAXN = (int)2e5+10;
const int INF = 0x3f3f3f3f;
const int mod = (int)1e9+7;

struct node{
    int l,r;
    int lazy,x;
    int len;
}t[MAXN<<2];
int n,m;
int cnt;
int f[MAXN], d[MAXN], sz[MAXN], son[MAXN], rk[MAXN], top[MAXN], id[MAXN];
vector<int> E[MAXN];
int a[MAXN];

void push_down(int o){
    if(t[o].lazy!=-1){
        t[ls].x=t[ls].len*t[o].lazy;
        t[rs].x=t[rs].len*t[o].lazy;
        t[ls].lazy=t[o].lazy;
        t[rs].lazy=t[o].lazy;
        t[o].lazy=-1;
    }
}
void build(int l,int r,int o) {
    t[o].l=l,t[o].r=r;t[o].len=r-l+1;
    t[o].lazy=-1; t[o].x=0;
    if(l==r) return ;
    int mid=(l+r)>>1;
    build(l,mid,ls); build(mid+1,r,rs);
    t[o].x=t[ls].x+t[rs].x;
}
inline void update(int l,int r, int o, int x) {
    if(t[o].l>=l&&t[o].r<=r) {
        t[o].x=t[o].len*x;
        t[o].lazy=x;
        return;
    }
    push_down(o);
    int mid=(t[o].l+t[o].r)>>1;
    if(r<=mid) update(l,r,ls,x);
    else if(l>mid) update(l,r,rs,x);
    else {
        update(l,mid,ls,x); update(mid+1,r,rs,x);
    }
    t[o].x = t[ls].x+t[rs].x;
}
inline int query(int l,int r,int o){
    if(t[o].l>=l&&t[o].r<=r) return t[o].x;
    push_down(o);
    int mid=(t[o].l+t[o].r)>>1;
    if(r<=mid) return query(l,r,ls);
    else if(l>mid) return query(l,r,rs);
    else
        return (query(l,mid,ls)+query(mid+1,r,rs));
}
// 树上操作-----------------------------
void dfs1(int u, int fa, int dep){
    f[u]=fa, d[u]=dep, sz[u]=1;
    rep(i,0,SZ(E[u])){
        int v=E[u][i];
        if(v==fa) continue;
        dfs1(v,u,dep+1);
        sz[u]+=sz[v];
        if(sz[v]>sz[son[u]]) son[u] = v;
    }
}
void dfs2(int u,int t){
    top[u]=t;
    id[u]=++cnt;
    rk[cnt]=u;
    if(!son[u]) return;
    dfs2(son[u],t);
    rep(i,0,SZ(E[u])){
        int v=E[u][i];
        if(v!=son[u]&&v!=f[u]) dfs2(v,v);
    }
}
void add_lca(int x,int y,int z) {
    while(top[x]!=top[y]) {
        if(d[top[x]]<d[top[y]]) swap(x,y);
        update(id[top[x]],id[x],1,z);
        x=f[top[x]];
    }
    if(d[x]>d[y]) swap(x,y);
    update(id[x],id[y],1,z);
}

int main() {
    cpp_io();
    cin>>n;
    repp(i,2,n) {
        int u; cin>>u;++u;
        E[u].pb(i); E[i].pb(u);
    }
    cin>>m;
    cnt=0;
    dfs1(1,0,1); dfs2(1,1);
    build(1,n,1);
    while(m--){
        string s;
        int u;int x1=t[1].x;
        cin>>s>>u;++u;
        if(s[0]=='i')  {
            add_lca(1,u,1);
            cout<<abs(x1-t[1].x)<<endl;
        }
        else {
            update(id[u],id[u]+sz[u]-1,1,0);
            cout<<abs(x1-t[1].x)<<endl;
        }
    }
    return 0;
}