PKU3321 Apple Tree - 树状数组

题目描述：

苹果树上有N个节点，苹果只能长在节点上，一个节点要么有一个苹果要么没有。给出树的结构，一种操作是将某个节点的苹果栽下或长出苹果；另一种操作是询问某个节点为根的子树总共有多少苹果。

分析：

题目变形很厉害，不过任然可以用树状数组来解决。

我们将节点重新编号，使得所有孩子节点的编号都小于父亲节点的编号，那么当询问子树总共有多少苹果时就是询问比该节点编号小的节点有多少苹果。也就是树状数组的模型了。

编号规则：照后续遍历的方式给节点编号，得到的编号一定是子树中最大的max，并且记录下该节点之下的最小编号min。那么转换为树状数组模型后，询问节点v时，则回答Sum(v.max)-Sum(v.min-1)。

当然，此题也可以编号之后用线段树来解决。

 

1. /*
2. PKU3321 Apple Tree
3. */
5. #include <stdio.h>
6. #include <memory.h>
7. #include <malloc.h>
9. #define clr(a) memset(a,0,sizeof(a))
11. #define N 100005
13. typedef struct LinkNode{
14.     int id;
15.     struct LinkNode* next;
16. }Link;
18. typedef struct{
19.     int index;
20.     int min;
21. }Index;
23. //vars
24. Link mem[N*2];
25. int pmem;
27. int n,m;
28. Link* a[N];
29. int e[N];
30. Index idx[N];
31. int nidx;
33. int b[N];
34. int c[N];
36. void addNode(int i,int j){
37.     Link *p = &mem[pmem++];
38.     p->id=j;
39.     p->next=a[i];
40.     a[i]=p;
41. }
43. void Traversal(int v){
44.     Link *p;
45.     e[v]=1;
46.     idx[v].min=nidx;
47.     
48.     p=a[v];
49.     while(p!=NULL){
50.         if(!e[p->id]){
51.             Traversal(p->id);
52.         }
53.         p=p->next;
54.     }
55.     idx[v].index=nidx++;
56. }
58. int lowbit(int x){
59.     return -x&x;
60. }
62. void Add(int i,int t){
63.     while(i<=n){
64.         c[i]+=t;
65.         i+=lowbit(i);
66.     }
67. }
69. int Sum(int i){
70.     int s=0;
71.     while(i>0){
72.         s+=c[i];
73.         i-=lowbit(i);
74.     }
75.     return s;
76. }
78. int main()
79. {
80.     while(scanf("%d",&n)!=EOF){
81.         int i,j,k;
82.         //input
83.         clr(a); pmem=0;
84.         
85.         for(k=0;k<n-1;k++){
86.             scanf("%d%d",&i,&j);
87.             addNode(i,j);
88.             addNode(j,i);
89.         }
90.         
91.         //search tree
92.         clr(e);
93.         nidx=1;
94.         Traversal(1);
95.         
96.         //init tree
97.         for(k=1;k<=n;k++){
98.             b[k]=1;
99.             Add(k,1);
100.         }
101.         
102.         //work
103.         char ch[5];
104.         int sum;
105.         scanf("%d",&m);
106.         while(m--){
107.             scanf("%s%d",ch,&k);
108.             i=idx[k].index;
109.             j=idx[k].min;
110.             
111.             if(ch[0]=='C'){  //change
112.                 b[i]=!b[i];
113.                 Add(i,b[i]?1:-1);
114.             }
115.             if(ch[0]=='Q'){  //inquiry
116.                 sum=Sum(i)-Sum(j-1);
117.                 printf("%d/n",sum);
118.             }
119.         }
120.     }
121.     
122.     return 0;
123. }