poj3321Apple Tree（树状数组）

http://poj.org/problem?id=3321

刚一看题以为要建一颗树 看了下讨论说dfs

这里dfs遍历时设的标号很好 一个low一个high 包含了以这一节点为根节点的子树结点的所有标号

 1 #include <iostream>
 2 #include<cstdio>
 3 #include<cstring>
 4 #include<algorithm>
 5 #include<stdlib.h>
 6 using namespace std;
 7 #define N 100010
 8 #define lowbit(x) (x&(-x))
 9 struct node
10 {
11     int u,v,next;
12 }ed[N<<1];
13 int head[N],t,low[N],g,high[N],vis[N],re[N],f[N];
14 void init()
15 {
16     t = 0;
17     memset(head,-1,sizeof(head));
18 }
19 void add(int u,int v)
20 {
21     ed[t].u = u;
22     ed[t].v = v;
23     ed[t].next = head[u];
24     head[u] = t;
25     t++;
26 }
27 void dfs(int u)
28 {
29     low[u] = ++g;
30     vis[u] = 1;
31     int i;
32     for(i = head[u] ; i != -1 ; i = ed[i].next)
33     {
34         int v = ed[i].v;
35         if(!vis[v])
36             dfs(v);
37     }
38     high[u] = g;
39 }
40 void add1(int i,int da)
41 {
42     while(i<=g)
43     {
44         re[i]+=da;
45         i+=lowbit(i);
46     }
47 }
48 int getsum(int i)
49 {
50     int sum = 0;
51     while(i)
52     {
53         sum+=re[i];
54         i-=lowbit(i);
55     }
56     return sum;
57 }
58 int main()
59 {
60     int i,j,k,n,m,a,b;
61     char s[5];
62     init();
63     scanf("%d",&n);
64     for(i = 1; i < n ; i++)
65     {
66         scanf("%d%d",&a,&b);
67         add(a,b);
68         add(b,a);
69     }
70     dfs(1);
71     for(i = 1 ; i <= g ; i++)
72     {
73         f[i] = -1;
74         add1(i,1);
75     }
76     scanf("%d",&m);
77     while(m--)
78     {
79         scanf("%s %d",s,&k);
80         if(s[0]=='Q')
81         {
82             printf("%d\n",getsum(high[k])-getsum(low[k]-1));
83         }
84         else
85         {
86             add1(low[k],f[k]);
87             f[k] = f[k]*-1;
88         }
89     }
90     return 0;
91 }
View Code

 

posted @ 2013-07-29 19:41  _雨  阅读(190)  评论(0)  编辑  收藏  举报