POJ 3321（树状数组）

考试的时候这道题没想出来，考完了在LCX的提示下想出来了。呵呵

树状数组。那怎么把这颗树拍平呢（转换成1维数组）？

其实很简单（当然是想到了才简单），从root开始dfs下去记录第一次访问u点时的时间戳为st[u]（下标），当访问完以u为根的所有子树以后要向上回溯的时候，在记录一个时间戳end[u]（下标），

这样，一棵树的所有顶点权值的和就是st[u]到end[u]的区间和，操作就和普通的树状数组一样了~

PS：注意读入，我这个蒟蒻把读入写错了，本机测试没事，上评测机就不行了，不知道为什么。。。

#include <cstdio>
#include <cstdlib>
#include <cstring>
#define N 1001000
using namespace std;
int head[N],next[N],to[N],d[N],cnt,num,st[N],end[N],n,m;
bool vis[N];
void add(int u,int v)
{
    to[cnt]=v; next[cnt]=head[u]; head[u]=cnt++;
}
void read()
{
    memset(head,-1,sizeof head);
    cnt=0;
    for(int i=1,a,b;i<n;i++)
    {
        scanf("%d%d",&a,&b);
        add(a,b); add(b,a);
    }
}
void dfs(int u)
{
    st[u]=++num;
    vis[u]=true;
    for(int i=head[u];~i;i=next[i])
        if(!vis[to[i]])
            dfs(to[i]);
    end[u]=num;
}
int lowbit(int k)
{
    return k&(-k);
}
void updata(int x,int nu)
{
    while(x<=num)
    {
        d[x]+=nu;
        x+=lowbit(x);
    }
}
int getsum(int x)
{
    int sum=0;
    while(x)
    {
        sum+=d[x];
        x-=lowbit(x);
    }
    return sum;
}
void go()
{
    memset(vis,0,sizeof vis);
    num=0;
    dfs(1);
    for(int i=1;i<=n;i++) updata(i,1);
    scanf("%d",&m);
    int a;char b;
    while(m--)
    {
        getchar();
        scanf("%c%d",&b,&a);
        if(b=='C') 
        {
            if(getsum(st[a])-getsum(st[a]-1)==1)
                updata(st[a],-1);
            else updata(st[a],1);
        }
        else printf("%d\n",getsum(end[a])-getsum(st[a]-1));
    }
}
int main()
{
    while(scanf("%d",&n)!=EOF)
    {
        read();
        go();
    }
    return 0;
}