poj3321Apple Tree（树状数组）

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

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

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

#include <iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<stdlib.h>
using namespace std;
#define N 100010
#define lowbit(x) (x&(-x))
struct node
{
    int u,v,next;
}ed[N<<1];
int head[N],t,low[N],g,high[N],vis[N],re[N],f[N];
void init()
{
    t = 0;
    memset(head,-1,sizeof(head));
}
void add(int u,int v)
{
    ed[t].u = u;
    ed[t].v = v;
    ed[t].next = head[u];
    head[u] = t;
    t++;
}
void dfs(int u)
{
    low[u] = ++g;
    vis[u] = 1;
    int i;
    for(i = head[u] ; i != -1 ; i = ed[i].next)
    {
        int v = ed[i].v;
        if(!vis[v])
            dfs(v);
    }
    high[u] = g;
}
void add1(int i,int da)
{
    while(i<=g)
    {
        re[i]+=da;
        i+=lowbit(i);
    }
}
int getsum(int i)
{
    int sum = 0;
    while(i)
    {
        sum+=re[i];
        i-=lowbit(i);
    }
    return sum;
}
int main()
{
    int i,j,k,n,m,a,b;
    char s[5];
    init();
    scanf("%d",&n);
    for(i = 1; i < n ; i++)
    {
        scanf("%d%d",&a,&b);
        add(a,b);
        add(b,a);
    }
    dfs(1);
    for(i = 1 ; i <= g ; i++)
    {
        f[i] = -1;
        add1(i,1);
    }
    scanf("%d",&m);
    while(m--)
    {
        scanf("%s %d",s,&k);
        if(s[0]=='Q')
        {
            printf("%d
",getsum(high[k])-getsum(low[k]-1));
        }
        else
        {
            add1(low[k],f[k]);
            f[k] = f[k]*-1;
        }
    }
    return 0;
}