题目大意:…蒟蒻英语语文都不好…codevs上有中文翻译qwq 上链接
这个题是dfs序的应用中初心者的题目。暑假学长就让我们做,然而当时我还不会什么树状数组,线段树也懒得打。这几天敲了几次树状数组的模板,就赶紧来填这个坑了。
dfs序据说是处理子树问题的神器。dfs序的性质就是一棵子树内的所有节点在序列中都是连续的(比如以u为根的子树,都在in[u]和out[u]之间)。我们在dfs过程中记录一下这两个时间戳,分别代表进入这个节点的时间戳和离开这个节点的时间戳。这样在树上的问题就可以转化成了区间问题。对于区间问题就可以用线段树或树状数组来很好的求解啦。
这里有个写的比较好的博文你们可以看下qwq链接
对于这个题来说,用dfs序把树上问题转化为区间问题以后,用树状数组维护单点修改和区间和就可以了(用线段树练习一的模板就可以了)
代码如下:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define lowbit(x) (x&-x)
#define SZ 200000+10
using namespace std;
int first[SZ],nxt[SZ<<1],in[SZ],out[SZ];
int tree[SZ];
int n,tot=0,tim=0;
struct edge{
int from,to,cost;
}es[SZ<<1];
bool vis[SZ],use[SZ];
char s[10];
void build(int ff,int tt)
{
es[++tot]=(edge){ff,tt};
nxt[tot]=first[ff];
first[ff]=tot;
}
void dfs(int u)
{
vis[u]=1;
in[u]=++tim;
for(int i=first[u];i!=-1;i=nxt[i])
{
int v=es[i].to;
if(!vis[v])
dfs(v);
}
out[u]=++tim;
}
void add(int x,int y)
{
while(x<=tim)
{
tree[x]+=y;
x+=lowbit(x);
}
}
int sum(int x)
{
int ans=0;
while(x)
{
ans+=tree[x];
x-=lowbit(x);
}
return ans;
}
int ask(int x,int y)
{
return sum(y)-sum(x-1);
}
int main()
{
memset(first,-1,sizeof(first));
scanf("%d",&n);
for(int i=1;i<n;i++)
{
int ff,tt;
scanf("%d%d",&ff,&tt);
build(ff,tt);
build(tt,ff);
}
dfs(1);
for(int i=1;i<=n;i++)
add(in[i],1);
int m;
scanf("%d",&m);
while(m--)
{
int a;
scanf("%s",&s);
if(s[0]=='C')
{
scanf("%d",&a);
if(!use[a])
{
use[a]=1;
add(in[a],-1);
}
else
{
use[a]=0;
add(in[a],1);
}
}
else
{
scanf("%d",&a);
printf("%d\n",ask(in[a],out[a]));
}
}
return 0;
}