题目链接:
http://codeforces.com/contest/877/problem/E
题解:
dfs序+线段树区间更新和区间求和
代码:
#include <cmath>
#include <cstdio>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;
#define met(a,b) memset(a,b,sizeof(a))
#define inf 0x3f3f3f3f
#define lchild root<<1,l,mid
#define rchild root<<1|1,mid+1,r
const int maxn = 2e5+10;
int L[maxn],R[maxn];
int tot;
vector<int>p[maxn];
int tree[maxn<<2],lazy[maxn<<2];
int n;
//dfs序
void dfs(int u,int fa)
{
L[u]=++tot;
for(int i=0;i<p[u].size();i++)
{
int v=p[u][i];
if(v==fa)
continue;
dfs(v,u);
}
R[u]=tot;
}
void push_up(int root)
{
tree[root]=tree[root<<1]+tree[root<<1|1];
}
void push_down(int root,int l,int r)
{
if(lazy[root])
{
int mid=(l+r)>>1;
tree[root<<1]=mid-l+1-tree[root<<1];
lazy[root<<1]^=1;
tree[root<<1|1]=r-(mid+1)+1-tree[root<<1|1];
lazy[root<<1|1]^=1;
lazy[root]=0;
}
}
void build(int root=1,int l=1,int r=n)
{
if(l==r)
{
// tree[root]=room[L[l]];
tree[root]=0;
return;
}
int mid=(l+r)>>1;
build(lchild),build(rchild);
push_up(root);
}
void updata(int L,int R,int root=1,int l=1,int r=n)
{
if(L<=l&&r<=R)
{
tree[root]=(r-l+1)-tree[root];
lazy[root]^=1;
return;
}
if(lazy[root])
push_down(root,l,r);
int mid=(l+r)>>1;
if(L<=mid)
updata(L,R,lchild);
if(R>mid)
updata(L,R,rchild);
push_up(root);
}
int query(int L,int R,int root=1,int l=1,int r=n)
{
if(L<=l&&r<=R)
return tree[root];
if(lazy[root])
push_down(root,l,r);
int mid=(l+r)>>1,res=0;
if(L<=mid)
res+=query(L,R,lchild);
if(R>mid)
res+=query(L,R,rchild);
return res;
}
int main()
{
scanf("%d",&n);
for(int i=2;i<=n;i++)
{
int x;
scanf("%d",&x);
p[x].push_back(i);
p[i].push_back(x);
}
met(tree,0),met(lazy,0);
met(L,0),met(R,0);
tot=0;
dfs(1,-1);
build();
for(int i=1;i<=n;i++)
{
int x;
scanf("%d",&x);
if(x==1)
updata(L[i],L[i]);
}
int m;
scanf("%d",&m);
while(m--)
{
string s;
int num;
cin>>s>>num;
if(s=="get")
printf("%d\n",query(L[num],R[num]));
else
updata(L[num],R[num]);
}
}