POJ3321:Apple Tree 树状数组 （一个点通过dfs变成两个值）

给出一个苹果树，每个节点一开始都有苹果

C X，如果X点有苹果，则拿掉，如果没有，则新长出一个

Q X，查询X点与它的所有后代分支一共有几个苹果

我们利用树状数组做有一个问题没法解决，那就是查询X及其他所有的后代。因为我们平时树状数组查询区间时，是两个点。现在只有一个点。并且我们不知道这个每个点从属于哪个点。所以我们需要dfs来赋值，对于每个点，我们要赋两个值，一个小的，一个大的，这个大的是包括了这个点所有子节点之后的值。也就是说大的-小的就是我们常用的树状数组做法。

          a

        /    \

     b       c

    /  \      /   \

 d   e   f     g

我们用S表示小值，L表示大值

赋值NUM = 1

我们从a开始搜，先让a的S=NUM(1)，,NUM+1,搜索第一个节点b，我们让b的S=NUM(2),NUM+1,在搜d，让d的S=NUM(3).之后我们d这个节点没有子节点了，让d的L=NUM（3）（也就是等于他的S。）。返回到b节点，我们在搜索e，NUM++，e的S=NUM（4），之后e没有可以搜的了，e的L=NUM（4），之后返回b，b没的搜了，b的L=NUM（4）。之后返回a，a在搜c

最后我们得到

                          a（1,7）

                                        

          b (2,4)                 c(5,7)

   

d(3,3)       e(4,4)        f(6,6)   g(7,7)

我们会发现，正好符合树状数组。

我们按照这个进行树状数字操作即可。

 

 

#include <cstdio>

#include <cstring>
#include <vector>
#define MAXN 100005
#define mem(a) memset(a, 0, sizeof(a))
using namespace std;


int TreeArray[MAXN], Left[MAXN], Right[MAXN], Fork[MAXN];
typedef vector<int> Ve;
vector<Ve>Edge(MAXN);
int N,M;
int key;


void init()//初始化数组和
{
    mem(Left);  mem(Right);
    mem(Fork);  mem(TreeArray);
    for(int i=0;i<MAXN;i++)Edge[i].clear();
}


void DFS(int node)//为每一个node添加一个左值和右值，表示这个节点所
{
    Left[node] = key;
    for(int i=0;i<Edge[node].size();i++)
    {
        key+=1;
        DFS(Edge[node][i]);
    }
    Right[node] = key;
}


int LowBit(int x)//返回的是2^k
{
    return x & (x ^ (x-1));
}


void Edit(int k, int num)//修改节点k，如果是添加一个，代入1，删除一个代入-1
{
    while(k <= N)
    {
        TreeArray[k] += num;
        k += LowBit(k);
    }
}


int GetSum(int k)//得到1...k的和
{
    int sum = 0;
    while(k>=1)
    {
        sum += TreeArray[k];
        k -= LowBit(k);
    }
    return sum;
}


void ReadDataAndDo()
{
    int a,b;
    char ch;
    for(int i=1;i<N;i++)//输入a，b把边存放在容器里面
    {
        scanf("%d%d", &a, &b);
        Edge[a].push_back(b);
    }
    key = 1;    DFS(1);//为每一个节点对应一个左边界和右边界，他自己就存放在左边界里面，而它的管辖范围就是左边界到右边界
    for(int i=1;i<=N;i++)
    {
        Fork[i] = 1;//最初每个Fork上都有一个苹果
        Edit(i,1);//同时更新树状数组的值
    }
    scanf("%d%*c", &M);
    for(int i=0;i<M;i++)
    {
        scanf("%c %d%*c", &ch, &b);
        if(ch == 'Q')//b的子树就是[Left[b], right[b]]
        {
            printf("%d\n", GetSum(Right[b]) - GetSum(Left[b]-1));
        }
        else
        {
            if(Fork[b]) Edit(Left[b],-1);//由于每个节点的编号就是它的左值，所以直接修改左节点
            else Edit(Left[b],1);
            Fork[b] = !Fork[b];//变为相反的状态
        }
    }
}


int main()
{
    while(~scanf("%d", &N))
    {
        init();
        ReadDataAndDo();
    }
    return 0;
}