POJ 3321 Apple Tree dfs+二叉索引树

题目：http://poj.org/problem?id=3321

动态更新某个元素，并且求和，显然是二叉索引树，但是节点的标号不连续，二叉索引树必须是连续的，所以需要转化成连续的，多叉树的形状已经建好，只要重新标号成连续的就行了。

感觉重新标号是这个题最难的地方，否则就是个纯水题了。。。

重新标号是看的别人的。。。用dfs遍历多叉树标号。

#include <stdio.h>
#include <stdlib.h>
#include <string.h>

int item = 0;
int c[100010], n;
int start[100010], end[100010];

struct Tree
{
    int num;
    struct Tree *next;
    Tree()
    {
        next = NULL;
        num = 0;
    }
}tree[100010];

int lowbit(int x)
{
    return x & (-x);
}

void add(int x, int y)
{
    while(x <= n)
    {
        c[x] += y;
        x += lowbit(x);
    }
}

int sum(int x)
{
    int ret = 0;
    while(x)
    {
        ret += c[x];
        x -= lowbit(x);
    }
    return ret;
}

int query(int x, int y)
{
    return sum(y) - sum(x-1);
}

void dfs(int x)
{
    start[x] = ++item;
    struct Tree *p = tree[x].next;
    while(p != NULL)
    {
        if(start[p->num] == 0)
            dfs(p->num);
        p = p->next;
    }
    end[x] = item;
}

int main()
{
    int u, v;
    scanf("%d", &n);
    memset(tree, 0, sizeof(tree));
    memset(start, 0, sizeof(start));
    memset(c, 0, sizeof(c));
    for(int i = 1; i < n; i++)
    {
        scanf("%d %d", &u, &v);
        struct Tree *p = new Tree;
        p->num = v;
        p->next = tree[u].next;
        tree[u].next = p;
    }
    dfs(1);
    for(int i = 1; i <= n; i++)
    {
        add(i, 1);
    }
    int q, z;
    char cmd[2];
    scanf("%d", &q);
    while(q--)
    {
        scanf("%s %d", cmd, &z);
        if(cmd[0] == 'Q')
        {
            printf("%d
", query(start[z], end[z]));
        }
        else
        {
            if(sum(start[z]) - sum(start[z]-1) == 1)
                add(start[z], -1);
            else add(start[z], 1);
        }
    }
    return 0;
}