POJ3321 Apple Tree dfs序+树状数组

原题地址
题意：给一颗根为1的树，节点数为n，一开始每个节点上有一颗苹果，现有两个操作：
1.Q x，输出x的子树上的苹果个数。
2.C x，如果x点有苹果就拿走，没有就长一个出来。
m次操作。

思路：用dfs序把树上的节点转换成区间形式，然后用树状数组或者线段树维护和查询，模板题。

#include <iostream>
#include <stdio.h>
#include<cmath>
#include <cstdio>
#include<algorithm>
#include <cstring>
#include <string>
#define LL long long
#define INF INT64_MAX
#define MOD 1000000007
#define stree SegTree[root]
#define lson SegTree[root << 1]
#define rson SegTree[root << 1 | 1]
using namespace std;
typedef pair<int,int>pa;
const int N = 5e5+7;
char s[105];
int cnt, k, vis[N], d[N], head[N], l[N], r[N];
struct node
{
    int to, next;
}edge[N];
void add(int u, int v)
{
    edge[k].to = v;
    edge[k].next = head[u];
    head[u] = k++;
}
void dfs(int x)
{
    l[x] = ++cnt;
    for(int i = head[x];i!=-1;i = edge[i].next)
        dfs(edge[i].to);
    r[x] = cnt;
}
int lowbit(int x) {return x&(-x);}
void update(int x, int val)
{
    for(int i = x;i <= cnt;i+=lowbit(i)) d[i] += val;
}
int sum(int x)
{
    int res = 0;
    for(int i = x;i; i-=lowbit(i)) res += d[i];
    return res;
}
int main()
{
    int t, n, m, x, k, u, v, q;
    scanf("%d", &n);
    memset(head, -1, sizeof(head));
    for(int i = 1;i < n;i++)
    {
        scanf("%d %d", &u, &v);
        add(u, v);
    }
    dfs(1);
    for(int i = 1;i <= n;i++)
    {
        vis[i] = 1;
        update(l[i], 1);
    }
    scanf("%d", &q);
    while(q--)
    {
        scanf("%s%d", s, &x);
        if(s[0]=='Q')
            printf("%d\n", sum(r[x])-sum(l[x]-1));
        else
        {
            if(vis[x]) update(l[x], -1);
            else update(l[x], 1);
            vis[x] ^= 1;
        }
    }
    return 0;
}