【链接】
poj3321
【题目大意】
给你一颗n个节点的树,树的根节点是1,有两个操作,分别是C x表示如果x节点的权值为1则修正为0,否则修正为1。Q x表示询问以x为根节点的子树的权值。
【解题报告】
直接用DFS序将树转化为区间,然后树状数组搞一下就可以了。
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=100005,maxm=200005;
int n,m,Time,tot,lnk[maxn],son[maxm],nxt[maxm],in[maxn],out[maxn];
bool vis[maxn];
struct BLT
{
int c[maxn];
void Clear(){memset(c,0,sizeof(c));}
int Lowbit(int x){return x&(-x);}
void Add(int x,int da){while (x<=n) c[x]+=da,x+=Lowbit(x);}
int Get(int x){int sum=0; while (x) sum+=c[x],x-=Lowbit(x); return sum;}
}tr;//树状数组
inline int Read()
{
int res=0;
char ch=getchar();
while (ch<'0'||ch>'9') ch=getchar();
while (ch>='0'&&ch<='9') res=res*10+ch-48,ch=getchar();
return res;
}
void Dfs(int x,int fa)
{
in[x]=++Time;
for (int j=lnk[x]; j; j=nxt[j])
if (son[j]!=fa) Dfs(son[j],x);
out[x]=Time;
}
void Add_e(int x,int y)
{
son[++tot]=y; nxt[tot]=lnk[x]; lnk[x]=tot;
}
int main()
{
freopen("3321.in","r",stdin);
freopen("3321.out","w",stdout);
n=Read(); Time=0;
for (int i=1,x,y; i<n; i++) x=Read(),y=Read(),Add_e(x,y),Add_e(y,x);
Dfs(1,0); m=Read(); tr.Clear();
for (int i=1; i<=n; i++) tr.Add(i,1);
memset(vis,0,sizeof(vis));
for (int i=1; i<=m; i++)
{
char ch=getchar();
while (ch!='Q'&&ch!='C') ch=getchar();
int x=Read();
if (ch=='Q') printf("%d\n",tr.Get(out[x])-tr.Get(in[x]-1));
else {if (vis[x]) tr.Add(in[x],1); else tr.Add(in[x],-1); vis[x]=1-vis[x];}
}
return 0;
}