【Codeforces Round #442 (Div. 2)E. Danil and a Part-time Job】 dfs序+线段树

链接：

Codeforces Round #442 (Div. 2)E. Danil and a Part-time Job

题意：

给你一颗有根树，树上每一个节点有一个灯，现在要支持两种操作，第一种操作是统计一颗子树内开着的灯个数。第二种操作是将一个子树内的所有灯状态改变(开灯->关灯，关灯->开灯)。

做法：

首先对于树的dfs序建立线段树，这样每一棵子树都能表示成一段区间，之后问题就转换为，区间改变01状态，区间求和。
这个问题我们可以通过lazy数组实现，首先我们统计出最初状态下每棵子树内灯的状况，将每棵子树内亮着的灯的个数表示为$sum$,对于区间更新，我们只需要用一个$add$表示这个区间是否被翻转，$add$每次和$1$进行异或，这表示如果偶数次翻转，区间就会回到最初的情况，之后我们只需要知道这个区间是否被翻转也就可以更新$sum$。

对于 $push\_down$函数，我们首先要更新子区间的$add$数组$(0->1,1->0)$，之后更新子区间的$sum$，这里我们要知道子区间内一共点的个数，也就是$(r-l+1)$，之后再更新$sum$即可。

对于$push\_up$函数，只需要合并子区间的和即可。

代码

#include<stdio.h>
#include<iostream>
#include<algorithm>
#include<string.h>
using namespace std;
const int maxn = 2e5+5;
int tim;
int t[maxn];
int sum[maxn],sum2[maxn];
int in[maxn],ou[maxn],dfn[maxn];
vector<int> G[maxn];
void dfs(int rt,int fa)
{
    in[rt]=++tim;
    dfn[tim]=rt;
    for(int i=0;i<G[rt].size();i++)
    {
        int to=G[rt][i];
        if(to==fa) continue;
        dfs(to,rt);
    }
    ou[rt]=tim;
}
struct T
{
    int l,r,mid;
    int add,sum;
}tree[maxn<<2];
void push_up(int rt)
{
    tree[rt].sum=tree[rt<<1].sum+tree[rt<<1|1].sum;
}
void push_down(int rt)
{
    if(tree[rt].add)
    {
        tree[rt<<1].add^=1;
        tree[rt<<1].sum=(tree[rt<<1].r-tree[rt<<1].l+1)-tree[rt<<1].sum;
        tree[rt<<1|1].add^=1;
        tree[rt<<1|1].sum=(tree[rt<<1|1].r-tree[rt<<1|1].l+1)-tree[rt<<1|1].sum;
        tree[rt].add=0;
    }
}
void build(int rt,int l,int r)
{
    tree[rt].add=0;
    tree[rt].l=l;
    tree[rt].r=r;
    if(l==r)
    {
        tree[rt].sum=t[dfn[l]];//dfs序的问题这里记得dfn
        return ;
    }
    int mid=tree[rt].mid=l+r>>1;
    build(rt<<1,l,mid);
    build(rt<<1|1,mid+1,r);
    push_up(rt);
}
void update(int rt,int l,int r)
{
    if(tree[rt].r<l||tree[rt].l>r) return ;
    if(tree[rt].l>=l&&tree[rt].r<=r)
    {
        tree[rt].sum=(tree[rt].r-tree[rt].l+1)-tree[rt].sum;
        tree[rt].add^=1;
        return ;
    }
    push_down(rt);
    if(tree[rt].mid>=l) update(rt<<1,l,r);
    if(tree[rt].mid<r) update(rt<<1|1,l,r);
    push_up(rt);
}
int query(int rt,int l,int r)
{
    if(tree[rt].r<l||tree[rt].l>r) return 0;
    if(tree[rt].l>=l&&tree[rt].r<=r) return tree[rt].sum;
    push_down(rt);
    int ans=0;
    if(tree[rt].mid>=l) ans+=query(rt<<1,l,r);
    if(tree[rt].mid<r) ans+=query(rt<<1|1,l,r);
    push_up(rt);
    return ans;
}
int main()
{
    int n,x;
    scanf("%d",&n);
    for(int i=2;i<=n;i++)
    {
        scanf("%d",&x);
        G[x].push_back(i);
        G[i].push_back(x);
    }
    for(int i=1;i<=n;i++) scanf("%d",&t[i]);
    dfs(1,-1);//获取dfs序
    build(1,1,n);
    int q;
    scanf("%d",&q);
    while(q--)
    {
        char op[4];
        int v;
        scanf("%s%d",op,&v);
        if(op[0]=='p') update(1,in[v],ou[v]);
        else printf("%d\n",query(1,in[v],ou[v]));
    }
    return 0;
}