给定一棵树,树上权值初始时为1。
给出两种操作:
1.将单点权值取反。
2.查询子树的权值和。
我们可以通过dfs将其映射到一个树状数组里,记录子树的起始位置以及终止位置。直接暴力改logn查询即可。
#include<stdio.h>
#include<vector>
using namespace std;
const int Maxn=2e5+5;
int n,m,Timer=0,tot=0;
int c[Maxn],in[Maxn],out[Maxn],f[Maxn];
bool b[Maxn];
struct edge{int nex,to;
}e[Maxn];
inline void add(int u,int v){e[++tot]=(edge){f[u],v};f[u]=tot;
}
void dfs(int x){in[x]=++Timer;for(int i=f[x];i;i=e[i].nex){int v=e[i].to;if(in[v]) continue;dfs(v);}out[x]=Timer;
}
inline int lowbit(int x){return x&-x;}
void update(int x,int v){ for(;x<=n;x+=lowbit(x))c[x]+=v; }
int query(int x){int sum=0;for(;x;x-=lowbit(x))sum+=c[x];return sum;}
int main(){scanf("%d",&n);int u,v;for(int i=1;i<n;++i){scanf("%d%d",&u,&v);add(u,v);add(v,u);}dfs(1);scanf("%d",&m);for(int i=1;i<=n+1;++i) update(i,1),b[i]=1;char s[4];for(int i=1;i<=m;++i){scanf("%s%d",s,&u);if(s[0]=='C') update(in[u],b[u]?-1:1),b[u]=1-b[u];else printf("%d\n",query(out[u])-query(in[u]-1));}return 0;
}