codeforces 877 problem E Danil and a Part-time Job 【dfs序 + 线段树区间异或修改】

传送门

//题意: 给定一颗有根(1 )树，树上每一个点起始都有一个值(0或1)，然后有两个操作，pow x把以 x为根的子树全部异或1， get x输出以x为根的子树中有多少个1.

//思路： 就是一个最裸的dfs序 + 线段树异或修改的，细节请看代码. 注意就是求一个区间中的1，那么就是区间长度-此时区间的值, 然后就是修改下lazy标记处就是了. 还有就是线段树中是dfs序的一些线段，所以在build过程中我们还需要知道此时的线段在树中对应的哪一个点，这样才能初始化.

AC Code

const int maxn = 2e5+5;
int cas=1;
int a[maxn];
int cnt,head[maxn];
int p1[maxn],p2[maxn];
int pre[maxn];
struct node
{
int to,next;
}e[maxn<<1];
void add(int u,int v)
{
e[cnt] = (node){v,head[u]};
head[u] = cnt++;
}
int ti;
void dfs_id(int u,int fa)
{
p1[u] = ++ti;
pre[ti] = u;
for(int i=head[u]; ~i ; i = e[i].next){
int to = e[i].to;
if(fa == to) continue;
dfs_id(to,u);
}
p2[u] = ti;
} //p1存的是第一次遇到时序号,p2是第二次遇到的序号.
//那么结点u及其子树存在的连续区间就是p1[u] - p2[u].
struct Tree{
int tl, tr, sum ,lazy;
void fun(int tot) {
lazy ^= 1;
sum = tot - sum;
}
} tree[maxn<<2];

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

void pushdown(int id,int l,int r)
{
if(tree[id].lazy){
int mid = (l+r)>>1;
tree[id<<1].fun(mid-l+1);
tree[id<<1|1].fun(r-mid);
tree[id].lazy = 0;
}
}

void build(int id,int l,int r)
{
tree[id].tl = l; tree[id].tr = r; tree[id].lazy = 0;
if(l == r){
tree[id].sum = a[pre[l]];
return ;
}
int mid = (l+r) >> 1;
build(id<<1,l,mid);
build(id<<1|1,mid+1,r);
pushup(id);
}

void update(int id,int st,int ed)
{
int l=tree[id].tl, r=tree[id].tr;
if(st <= l && r <= ed){
tree[id].fun(r-l+1);
return ;
}
pushdown(id,l,r);
int mid = (l+r) >> 1;
if(st <= mid) update(id<<1,st,ed);
if(ed > mid) update(id<<1|1,st,ed);
pushup(id);
}
int ans;
void query(int id,int ql,int qr)
{
int l = tree[id].tl , r = tree[id].tr;
if(ql <= l && r <= qr) {
ans += tree[id].sum;
return ;
}
pushdown(id,l,r);
int mid = (l+r) >> 1;
if(ql <= mid) query(id<<1,ql,qr);
if(qr > mid) query(id<<1|1,ql,qr);
}

void solve()
{
int n; scanf("%d",&n);
ti = 0; Fill(pre,0);
cnt = 0; Fill(head,-1);
for(int i=2;i<=n;i++){
int u;
scanf("%d",&u);
add(i,u); add(u,i);
}
for(int i=1;i<=n;i++) {
scanf("%d",&a[i]);
}
dfs_id(1,-1);
build(1,1,n);
int q; scanf("%d",&q);
while(q--){
string s; cin >> s;
int u; scanf("%d",&u);
if(s == "get"){
ans = 0;
query(1,p1[u],p2[u]);
printf("%d\n",ans);
}
else{
update(1,p1[u],p2[u]);
}
}
}