877E - Danil and a Part-time Job（dfs序＋线段树）

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define pb push_back
#define ls rt<<1,l,m
#define rs rt<<1|1,m+1,r
#define mem(a,b) memset(a,b,sizeof(a)) 

const int N=2e5+5;
int tree[4*N];
int lazy[4*N];
int a[N];
int in[N];
int out[N];
vector<int>g[N];
void dfs(int o,int u,int &x)
{
    in[u]=x;
    for(int i=0;i<g[u].size();i++)if(g[u][i]!=o)dfs(u,g[u][i],++x);
    out[u]=x; 
}
void push_up(int rt)
{
    tree[rt]=tree[rt<<1]+tree[rt<<1|1];
}
void push_down(int rt,int len)
{
    lazy[rt<<1]^=lazy[rt];//对lazy数组很巧妙地运用，两次变换相当于没变 
    lazy[rt<<1|1]^=lazy[rt];
    tree[rt<<1]=(len-(len>>1))-tree[rt<<1];
    tree[rt<<1|1]=(len>>1)-tree[rt<<1|1];
    lazy[rt]=0;
}
void build(int rt,int l,int r)
{
    lazy[rt]=0;
    if(l==r)
    {
        tree[rt]=a[l];
        return ;
    }
    int m=(l+r)>>1;
    build(ls);
    build(rs);
    push_up(rt);
}
void Update(int L,int R,int rt,int l,int r)
{
    if(L<=l&&r<=R)
    {
        lazy[rt]^=1;
        tree[rt]=r-l+1-tree[rt];
        return ;
    }
    if(lazy[rt])push_down(rt,r-l+1);
    int m=(l+r)>>1;
    if(L<=m)Update(L,R,ls);
    if(R>m)Update(L,R,rs);
    push_up(rt);
}
int query(int L,int R,int rt,int l,int r)
{
    if(L<=l&&r<=R)return tree[rt];
    if(lazy[rt])push_down(rt,r-l+1);
    int m=(l+r)>>1;
    int ans=0;
    if(L<=m)ans+=query(L,R,ls);
    if(R>m)ans+=query(L,R,rs);
    return ans;
} 

int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    int n,p,q,t;
    string s;
    cin>>n;
    for(int i=2;i<=n;i++)
    {
        cin>>p;
        g[p].pb(i);
        g[i].pb(p);
    }
    int x=1;
    dfs(0,1,x);
    for(int i=1;i<=n;i++)cin>>a[in[i]];
    build(1,1,n);
    cin>>q;
    while(q--)
    {
        cin>>s>>t;
        if(s=="get")cout<<query(in[t],out[t],1,1,n)<<endl;
        else Update(in[t],out[t],1,1,n); 
    }
    return 0;
}