POJ 3321 Apple Tree （树状数组）

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

题意：给出一颗苹果树，开始每个树叉上都有 一个苹果，苹果树总是以1为根，Q 1表示1树枝上有多少苹果，C 2表示更新树枝2上的苹果，以前有苹果变成没有，以前没有变成有，树杈上的苹果数不超过1

分析：插入，删除频繁，还需要算和，10^5的数据量，用线段树或树状数组，但是给出的是一棵树，不在一条线上。。。有时候不能看出什么是区间，就要仔细观察，造出“区间”，

如果从1根节点出发，dfs遍历树一遍，那遍历每个节点的时间是一条线，把时间划成区间。。。每个节点有两个时间戳，low[]表示开始时间，high[]表示结束时间。那么这个树枝上苹果的个数就是开始时间和结束时间之间的苹果

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<string>
#define nMAX 100005
using namespace std;
int low[nMAX],high[nMAX],c[nMAX],head[nMAX];
int n,m,s_edge,times;
bool vs[nMAX],has[nMAX];
struct Edge
{
    int u,v,nxt;
}edge[nMAX*2];
struct Tree
{
    int l,r;
}tree[nMAX*3+100];
void addedge(int u,int v)
{
    s_edge++;
    edge[s_edge].v=v;
    edge[s_edge].nxt=head[u];
    head[u]=s_edge;
}
void dfs(int u)
{
    low[u]=++times;
    vs[u]=1;
    for(int e=head[u];e;e=edge[e].nxt)
    {
        int v=edge[e].v;
        if(!vs[v]) dfs(v);
    }
    high[u]=times;
}

int lowbit(int x)
{
    return x&(-x);
}
void add(int i ,int num)
{
    while(i<=n)
    {
        c[i]+=num;
        i+=lowbit(i);

    }
}
int sum(int i)
{
    int ans=0;
    while(i>=1)
    {
        ans+=c[i];
        i-=lowbit(i);
    }
    return ans;

}
int main()
{
    char ch;
    int i,j,k;
    while(~scanf("%d",&n))
    {
        memset(head,0,sizeof(head));
        s_edge=0;
        for(k=1;k<n;k++)
        {
            scanf("%d%d",&i,&j);
            addedge(i,j);
            addedge(j,i);
        }
        memset(c,0,sizeof(c));
        for(i=1;i<=n;i++)//每个树枝都有apple
            add(i,1);
        times=0;
        memset(vs,0,sizeof(vs));
        dfs(1);
        memset(has,0,sizeof(has));//都有
        scanf("%d",&m);
        while(m--)
        {
            getchar();
            scanf("%c%d",&ch,&i);
            if(ch=='Q')
            {
                int ans=sum(high[i])-sum(low[i]-1);
                printf("%d\n",ans);
            }
            else
            {
                if(!has[i]){add(low[i],-1);has[i]=1;}
                else {add(low[i],1);has[i]=0;}

            }
        }
    }
    return 0;
}