[树状数组] poj 3321 Apple Tree

题目分析

    在树上做单点更新、区间查询，关键是要找一个“序”，使得以某点i为根的子树在一个连续区间[i,l+len-1]内（len为该子树包含根的节点个数，先用树形dp预处理一下），于是可以用中序遍历定序。
    更新时查看该点是否有苹果（开个布尔数组记录一下），有则-1，无则+1。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#include<cmath>
#include<algorithm>
#include<map>
using namespace std;
const int maxn=1e5+5;
int n,m;
struct Edge{int to,nt;}tree[2*maxn];
int nume;
int head[maxn];
int dp[maxn];//root i;point number
bool vis[maxn];
void build(int u,int v)
{
    tree[nume].to=v;
    tree[nume].nt=head[u];
    head[u]=nume++;
}
int bit[maxn];
bool exs[maxn];
int pos[maxn],cnt,ori[maxn];
void dfs(int cur)
{
    vis[cur]=true;
    pos[cur]=cnt;
    ori[cnt]=cur;
    cnt++;
    dp[cur]=1;
    int son;
    for(int i=head[cur];~i;i=tree[i].nt)
    {
        son=tree[i].to;
        if(vis[son]) continue;
        dfs(son);
        dp[cur]+=dp[son];
    }
}
inline int lowbit(int x){return x&-x;}
void add(int x,int val)
{for(int i=x;i<=n;i+=lowbit(i)) bit[i]+=val;}
int getsum(int x)
{
    int res=0;
    for(int i=x;i>0;i-=lowbit(i)) res+=bit[i];
    return res;
}
int main()
{
    cin>>n;
    nume=0;
    memset(head,-1,sizeof head);
    memset(bit,0,sizeof bit);
    memset(vis,0,sizeof vis);
    memset(exs,true,sizeof exs);
    char op[2];
    int u,v;
    cnt=1;
    for(int i=1;i<n;i++)
    {
        scanf("%d%d",&u,&v);
        build(u,v);
        build(v,u);
    }
    for(int i=1;i<=n;i++) add(i,1);
    dfs(1);
    cin>>m;
    for(int i=0;i<m;i++)
    {
        scanf("%s%d",op,&u);
        u=pos[u];
        if(op[0]=='C')
        {
            exs[u]=!exs[u];
            if(exs[u]) add(u,1);
            else add(u,-1);
        }
        else printf("%d\n",getsum(u+dp[ori[u]]-1)-getsum(u-1));
    }
    return 0;
}