树状数组:对区间进行:①增②改③查
思路:该题就是求解区间的值(该区间苹果的个数),因此 我们需要处理树杈的问题,当前的树杈是由哪个树杈引出来的,我们需要找到起始树杈(当前树杈)和
结束的树杈,然后,在用树状数组去维护,进行查找,修改。用DFS()对当前树杈的起始点和终止点进行标号,类似二叉树的遍历(该题是多叉树)
#include<iostream>
#include<algorithm>
#include<string.h>
#include<cstdio>
using namespace std;
const int maxn=100010;
int num[maxn];
int st[maxn],ed[maxn];
struct node//利用广义表的结构
{
int data;
node *next;
} Node[maxn];
int N,M,t;
char c;
void DFS(int pox)//多叉树搜索。
{
st[pox]=++t;
node *tem=Node[pox].next;
while(tem)
{
if(st[tem->data]==0)
{
DFS(tem->data);
}
tem=tem->next;
}
ed[pox]=t;
}
int lowbit(int x)
{
return -x&x;
}
int get_sum(int pox)
{
int sum=0;
while(pox>0)
{
sum+=num[pox];
pox-=lowbit(pox);
}
return sum;
}
void updata(int pox,int nu)
{
while(pox<=N)
{
num[pox]+=nu;
pox+=lowbit(pox);
}
}
int main()
{
int a,b;
int tem;
while(scanf("%d",&N)!=EOF)
{
memset(num,0,sizeof(num));
t=0;
for(int i=1; i<N; i++)
{
scanf("%d%d",&a,&b);
node *p=new node();
p->data=b;
p->next=Node[a].next;
Node[a].next=p;
node *q=new node();
q->data=a;
q->next=Node[b].next;
Node[b].next=q;
}
DFS(1);
// printf("st[2]=%d\n",st[2]);
for(int i=1; i<=N; i++)
{
updata(i,1);
}
for(int i=1; i<=N+1; i++)
scanf("%d",&M);
for(int i=0; i<M; i++)
{
cin>>c>>tem;
if(c=='Q')
{
printf("%d\n",get_sum(ed[tem])-get_sum(st[tem]-1));
}
else
{
int sum=get_sum(st[tem])-get_sum(st[tem]-1);
if(sum==1)
updata(st[tem],-1);
else
updata(st[tem],1);
}
}
}
return 0;
}