Description
你家门前种了一棵苹果树,每年秋天,树上都结满了苹果,你非常喜欢吃苹果,所以一直精心照料着苹果树。 苹果树有N个分叉,分叉之间由枝干相连,你把分叉用1到N来标记,树根必须记为1。苹果长在分叉处,而且一个分叉最多只能同时结一个苹果,也就是说不可能有超过一个苹果同时长在分叉处。你想知道某个子树中一共有多少个苹果。 上面的问题不难,但现在的问题是有时你会去摘苹果,有时又长出新的苹果,那你还能做吗?
Input
输入文件第一行是一个整数N(N<=100000),表示树中分叉的数目。 接下来N-1行每行两个整数u和v,表示分叉u和分叉v有一个枝干相连。 下一行是一个整数M(M<=100000) 接下来M行,每行或者是“C x”表示x分叉处的状态发生改变,如果原来有,就被摘下来,原来没有就长出新的苹果;或者是“Q x”表示询问以分叉x为根的子树中苹果的数量。 注意起初每个分叉都有一个苹果。
Output
对于每个询问输出对应的数量。
Sample Input
3
1 2
1 3
3
Q 1
C 2
Q 1
Sample Output
3
2
Sulotion
这题是一道典型的线段树,需要单点修改和区间求和。
但一眼看去,貌似不知道区间的左右端点
但其实在DFS的时候就可以将点处理标号,一个点的子树的标号一定是连续的,
那么对于一个点,存住它子树所在的区间,即可进行线段树操作。
Code
#include<cstdio>
using namespace std;
const int N=100001;
int tot;
int first[N],next[N<<1],en[N<<1];
int mx[N],g[N],f[N<<2];
inline int read()
{
int X=0; char ch=0;
while(ch<'0' || ch>'9') ch=getchar();
while(ch>='0' && ch<='9') X=(X<<3)+(X<<1)+ch-'0',ch=getchar();
return X;
}
inline void insert(int x,int y)
{
next[++tot]=first[x];
first[x]=tot;
en[tot]=y;
}
inline void dfs(int x,int y)
{
mx[x]=g[x]=++tot;
for(int i=first[x];i;i=next[i])
if(en[i]!=y)
{
dfs(en[i],x);
mx[x]=mx[en[i]];
}
}
inline void make(int v,int l,int r)
{
if(l==r)
{
f[v]=1;
return;
}
int mid=(l+r)>>1;
make(v<<1,l,mid);
make(v<<1|1,mid+1,r);
f[v]=f[v<<1]+f[v<<1|1];
}
inline void change(int v,int l,int r,int x)
{
if(l==r)
{
f[v]=f[v]^1;
return;
}
int mid=(l+r)>>1;
if(x<=mid) change(v<<1,l,mid,x); else change(v<<1|1,mid+1,r,x);
f[v]=f[v<<1]+f[v<<1|1];
}
inline int query(int v,int l,int r,int x,int y)
{
if(l==x && r==y) return f[v];
int mid=(l+r)>>1;
if(y<=mid) return query(v<<1,l,mid,x,y);
if(x>mid) return query(v<<1|1,mid+1,r,x,y);
return query(v<<1,l,mid,x,mid)+query(v<<1|1,mid+1,r,mid+1,y);
}
int main()
{
int n=read();
for(int i=1;i<n;i++)
{
int x=read(),y=read();
insert(x,y);
insert(y,x);
}
dfs(1,tot=0);
make(1,1,n);
int m=read();
while(m--)
{
char ch=getchar();
while(ch!='Q' && ch!='C') ch=getchar();
int x=read();
if(ch=='C') change(1,1,n,g[x]); else
printf("%d\n",query(1,1,n,g[x],mx[x]));
}
return 0;
}