思路:
我的思路是借鉴别人的。传送门:https://blog.csdn.net/libin56842/article/details/46634095
思路还是很巧妙,特别是建立区间。我们通过DFS来给所有的苹果编号,DFS求出每个节点索包含的区间,即开始位置和结束位置。对于某一个节点而言,我们用DFS遍历完它所有的后代才返回的(递归函数),所以,这个点对应的结束位置是它后代的最大编号。开始位置和结束位置就是一个区间,而C操作就是单点更新,Q操作就是区间查询,所以可以用树状数组来做。
AC code:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#include<set>
#include<vector>
#include<cmath> //因为涉及到大量查询区间和 以及 修改值操作,所以需要使用树状数组
using namespace std;
const int maxn = 1e5 + 10;
int start1[maxn],end1[maxn];//记录开始,结尾
bool sign[maxn];
//int sign[maxn];
int tree[maxn];
int n = 0,cnt = 0,m = 0;
vector< vector<int > > at(maxn);//预分配空间
int lowbit(int x){
return x & ( -x );
}
void update(int loc,int val){//单点更新
for(int i = loc;i <= n;i += lowbit(i)){
tree[i] += val;
}
}
int get_sum(int loc){//1-loc区间的和
int sum = 0;
for(int i = loc;i > 0;i -= lowbit(i)){//i > 0而不是i >= 0,否则会陷入死循环
sum += tree[i];
}
return sum;
}
void dfs(int u){//建立区间(抽象,很关键) ,每个节点管理的区间与他本身的编号无关
start1[u] = cnt;
for(int i = 0;i < (int)at[u].size();++i){
++cnt;
dfs(at[u][i]);
}
end1[u] = cnt;
}
int main(){
while(scanf("%d",&n) != EOF && n){
at.clear();
at.resize(maxn);
for(int i = 0;i < n-1;++i){//建立邻接表
int u =0,v = 0;
scanf("%d %d",&u,&v);
at[u].push_back(v);
}
//DFS处理
cnt = 1;
dfs(1);
//初始化
memset(tree,0,sizeof(tree));
for(int i = 1;i <= n;++i){
sign[i] = true;
//sign[i] = 1;
update(i,1);
}
//操作
scanf("%d",&m);
for(int i = 0;i < m;++i){
int loc = 0;
char cmd[10];
scanf("%s %d",cmd,&loc);
if(cmd[0] == 'C'){//单点更新
update(start1[loc], sign[loc] == true ? -1 : 1);
//update(start1[loc], sign[loc] == 1 ? -1 : 1);
sign[loc] = !(sign[loc]);
//sign[loc] = 1 - sign[loc];
}else{//区间求和
printf("%d\n",get_sum(end1[loc]) - get_sum(start1[loc] - 1));
}
}
}
return 0;
}