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

题目链接：https://codeforces.com/problemset/problem/877/E#

题意

        给你一棵n个结点的树，每个结点代表一盏灯，灯只有亮和灭两种状态。现在只有两种操作get和pow，其中get操作是获取某个结点及其子树内的所有结点中灯亮的个数，而pow操作是将某个结点及其子树中的所有节点的灯的亮灭状态转换（亮变成灭，灭变成亮）。

思路

        因为修改灯的状态是修改其子树内所有结点的状态，所以可以采用dfs序的方式建树，这样每一棵子树都能表示成一段连续的区间，之后问题就转换为区间改变01状态，区间求和。
        对于区间更新，需要记录一个lazy值表示区间内pow操作的次数，因为经过偶数次反转区间灯的状态不变，所以没进行一次pow操作只需对lazy的值异或1即可。
        对于PushUp函数只需上推子结点的sum和即可。而对于PushDown函数，需要修改lazy值，并且需要更新子节点的sum值，sum更新后值为区间内所有灯的个数减去更新前的sum值，这就实现了区间内灯亮状态的反转。

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int INF = 0x3f3f3f3f;
const int maxn=2e5+10;

int tim;
vector<int> G[maxn];
int t[maxn];
//in数组记录的是线段树上某个结点及其子树对应的dfs序的区间在dfn中该区间开始的下标。
//ou数组记录的是线段树上某个结点及其子树对应的dfs序的区间在dfn中该区间结束的下标。
//dfn数组记录的是线段树的dfs序
int in[maxn],ou[maxn],dfn[maxn];

struct node{
	int l,r;
	int lazy;
	int sum;//记录区间内亮着的灯的个数
}tree[maxn<<2];

void dfs(int rt,int fa)
{
	in[rt]=++tim;
	dfn[tim]=rt;
	for(auto to:G[rt]){
		if(to==fa) continue;
		dfs(to,rt);
	}
	ou[rt]=tim;
}

void PushUp(int rt)
{
	tree[rt].sum=tree[rt<<1].sum+tree[rt<<1|1].sum;
}

void PushDown(int rt)
{
	if(tree[rt].lazy){
		tree[rt<<1].lazy^=1;
		tree[rt<<1|1].lazy^=1;
		tree[rt<<1].sum=(tree[rt<<1].r-tree[rt<<1].l+1)-tree[rt<<1].sum;
		tree[rt<<1|1].sum=(tree[rt<<1|1].r-tree[rt<<1|1].l+1)-tree[rt<<1|1].sum;
		tree[rt].lazy=0;
	}
}
//根据dfs序建树
void Build(int l,int r,int rt)
{
	tree[rt].l=l;
	tree[rt].r=r;
	tree[rt].lazy=0;
	if(l==r){
		tree[rt].sum=t[dfn[l]];
		return;
	}
	int mid=(tree[rt].l+tree[rt].r)>>1;
	Build(l,mid,rt<<1);
	Build(mid+1,r,rt<<1|1);
	PushUp(rt);
}

void Update(int l,int r,int rt)
{
	if(l<=tree[rt].l&&tree[rt].r<=r){
		tree[rt].lazy^=1;
		tree[rt].sum=tree[rt].r-tree[rt].l+1-tree[rt].sum;
		return;
	}
	PushDown(rt);
	int mid=(tree[rt].l+tree[rt].r)>>1;
	if(r<=mid) Update(l,r,rt<<1);
	else if(l>mid) Update(l,r,rt<<1|1);
	else{
		Update(l,mid,rt<<1);
		Update(mid+1,r,rt<<1|1);
	}
	PushUp(rt);
}

int Query(int l,int r,int rt)
{
	if(l<=tree[rt].l&&tree[rt].r<=r){
		return tree[rt].sum;
	}
	PushDown(rt);
	int mid=(tree[rt].l+tree[rt].r)>>1;
	int ans=0;
	if(r<=mid) ans+=Query(l,r,rt<<1);
	else if(l>mid) ans+=Query(l,r,rt<<1|1);
	else{
		ans+=Query(l,mid,rt<<1);
		ans+=Query(mid+1,r,rt<<1|1);
	}
	PushUp(rt);
	return ans;
}

int main()
{
	int n;
	scanf("%d",&n);
	for(int i=2;i<=n;i++){
		int p;
		scanf("%d",&p);
		G[i].push_back(p);
		G[p].push_back(i);
	}
	for(int i=1;i<=n;i++) scanf("%d",&t[i]);
	dfs(1,-1);
	Build(1,n,1);
	int q;
	scanf("%d",&q);
	while(q--){
		char op[10];
		int v;
		scanf("%s %d",op,&v);
		if(strcmp(op,"get")==0) printf("%d\n",Query(in[v],ou[v],1));
		else Update(in[v],ou[v],1);
	}
	return 0;
}