Apple Tree POJ

来源：互联网发布：淘宝店banner在线制作编辑：程序博客网时间：2024/05/12 04:40

转载别人的http://www.cnblogs.com/gj-Acit/p/3236843.html

题目大意级是说，给你一颗树，最初每个节点上都有一个苹果，有两种操作：修改（即修改某一个节点，修改时这一个节点苹果从有到无，或从无到有）和查询（查询某一个节点他的子树上有多少个苹果）。
由于此题数据比较大(N<=10^5)，而且不是标准的二叉树，所以这里我们队每一个节点重新编号，另外为每一个节点赋一个左值和一个右值，表示这个节点的管辖范围。



上图也就是DFS搜索的时候做标记的过程，这样新的编号为1~6的节点所管辖的范围分别就是[1,6]  [2,4]   [3,3]  [4,4]  [5,6]  [6,6]，其中左边的是左值，右边的是右值，节点1的区间是[1,6]，正好这棵子树有6个节点，其他也一样
那我们吧新的节点放进树状数组时

自己敲了一遍代码，但是编译器说错误。。。

#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;}

0 0