【POJ 3321】Apple Tree

有n个节点以1为根节点的树，给你树的边关系u-v，一开始每个节点都有一个苹果，接下来有两种操作，C x改变节点x的苹果状态，Q x查询x为根的树的所有苹果个数。

 

求出树的dfs序，st[i]保存i的进入时间戳，ed[i]保存i的退出时间戳，则st[i]到ed[i]就是子树节点的对应时间戳。

每个节点打了两次时间戳，其中ed[i]会等于最后访问的一个子节点的st[i]。

于是用线段树/树状数组的单点修改和区间求和就可以解决。

#include<cstdio>
#define N 100005
using namespace std;
int n,m,id,cnt;
int q[N],st[N],ed[N],head[N];
int rl[N<<],rr[N<<],sum[N<<];
struct edge{
    int to,next;
}e[N<<];
void add(int u,int v){
    e[++cnt]=(edge){v,head[u]};
    head[u]=cnt;
}
void dfs(int x,int fa){
    st[x]=++id;
    for(int i=head[x];i;i=e[i].next)
        if(e[i].to!=fa)
            dfs(e[i].to,x);
    ed[x]=id;
}
void build(int k,int l,int r){
    rl[k]=l;rr[k]=r;
    if(l==r){
        sum[k]=;
        return;
    }
    int mid=l+r>>;
    build(k<<,l,mid);
    build(k<<|,mid+,r);
    sum[k]=sum[k<<]+sum[k<<|];
}
int query(int k,int a,int b){
    int l=rl[k],r=rr[k];
    if(a<=l&&r<=b) return sum[k];
    if(b<l||a>r)return ;
    return query(k<<,a,b)+query(k<<|,a,b);
}
void modify(int k,int x){
    int l=rl[k],r=rr[k],mid=(l+r)>>;
    if(l==r){
        sum[k]^=;
        return;
    }
    if(x<=mid)modify(k<<,x);
    else modify(k<<|,x);
    sum[k]=sum[k<<]+sum[k<<|];
}
int main(){
    scanf("%d",&n);
    for(int i=;i<n;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        add(u,v);
        add(v,u);
    }
    dfs(,);
    scanf("%d",&m);
    build(,,n);
    for(int i=;i<=m;i++){
        char op;
        scanf(" %c",&op);
        int x;
        scanf("%d",&x);
        if(op=='Q')
            printf("%d\n",query(,st[x],ed[x]));
        else
            modify(,st[x]);
    }
    return ;
}

树状数组

#include<cstdio>
#define N 100005
using namespace std;
int n,m,id,cnt;
int q[N],st[N],ed[N],head[N];
int cal[N],a[N];
struct edge{
    int to,next;
}e[N<<];
void add(int u,int v){
    e[++cnt]=(edge){v,head[u]};
    head[u]=cnt;
}
void dfs(int x,int fa){
    st[x]=++id;
    for(int i=head[x];i;i=e[i].next)
        if(e[i].to!=fa)
            dfs(e[i].to,x);
    ed[x]=id;
}
int getsum(int x){
    int s=;
    for(;x;x-=x&(-x))s+=cal[x];
    return s;
}
void update(int v,int x){
    for(;x<=n;x+=x&(-x))cal[x]+=v;
}
int main(){
    scanf("%d",&n);
    for(int i=;i<n;i++){
        int u,v;
        scanf("%d%d",&u,&v);
        add(u,v);add(v,u);
    }
    dfs(,);
    scanf("%d",&m);
    for(int i=;i<=n;i++)update(,i);
    for(int i=;i<=m;i++){
        char op;
        scanf(" %c",&op);
        int x;
        scanf("%d",&x);
        if(op=='C'){
            update(a[x]?:-,st[x]);
            a[x]^=;
        }
        else
            printf("%d\n",getsum(ed[x])-getsum(st[x]-));
    }
    return ;
}