【DFS序+单点修改区间求和】POJ 3321 Apple Tree

poj.org/problem?id=3321

【题意】

给一棵n个节点的树，每个节点开始有一个苹果，m次操作

1.将某个结点的苹果数异或 1

2.查询一棵子树内的苹果数

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include<string>
 #include<cmath>
 #include<algorithm>
 using namespace std;
 typedef long long ll;

 const int maxn=1e5+;
 const int maxm=*maxn;
 int n;
 struct edge
 {
     int to;
     int nxt;
 }e[maxm];
 int head[maxn];
 int tot;
 int l[maxn],r[maxn];
 int tree[maxm];
 bool vis[maxn];
 int cid;
 int lowbit(int x)
 {
     return x&(-x);
 }

 void add(int k,int x)
 {
     while(k<=cid)//注意是cid,不是n
     {
         tree[k]+=x;
         k+=lowbit(k);
     }
 }

 int query(int k)
 {
     int ans=;
     while(k)
     {
         ans+=tree[k];
         k-=lowbit(k);
     }
     return ans;
 }
 void init()
 {
     memset(head,-,sizeof(head));
     tot=;
     cid=;
     memset(vis,false,sizeof(vis));
 }

 void addedge(int u,int v)
 {
     e[tot].to=v;
     e[tot].nxt=head[u];
     head[u]=tot++;
 }

 void dfs(int u,int pa)
 {
     l[u]=++cid;
     for(int i=head[u];i!=-;i=e[i].nxt)
     {
         int v=e[i].to;
         if(v==pa) continue;
         dfs(v,u);
     }
     r[u]=++cid;
 }

 int main()
 {
     while(~scanf("%d",&n))
     {
         init();
         int u,v;
         for(int i=;i<=n-;i++)
         {
             scanf("%d%d",&u,&v);
             addedge(u,v);
             addedge(v,u);
         }
         dfs(,-);
         for(int i=;i<=n;i++)
         {
             add(l[i],);//单点修改
         }
         int q;
         scanf("%d",&q);
         char str[];
         int x;
         while(q--)
         {
             scanf("%s%d",str,&x);
             if(str[]=='Q')
             {
                 int ans=query(r[x])-query(l[x]-);
                 printf("%d\n",ans);
             }
             else
             {
                 if(!vis[x])
                 {
                     add(l[x],-);
                     vis[x]=true;
                 }
                 else
                 {
                     add(l[x],);
                     vis[x]=false;
                 }
             }
         }
     }
     return ;
  }