poj 3321 Apple Tree (Binary Index Tree)

3321 -- Apple Tree

　　做回之前个人赛超时到比赛结束的题。

　　这题就是一个简单的树形结构转化为线性结构，然后套用树状数组求区间和。不过不知道什么原因，用vector构建邻接表在dfs的时候超时了，转成链表就360ms过了。对于才1e5条边这么小的规模，我还是第一次遇见STL会超时的。

代码如下：

#include <cstdio>
#include <iostream>
#include <algorithm>
#include <vector>
#include <cstring>

using namespace std;

#define REP(i, n) for (int i = 0; i < (n); i++)
#define REP_1(i, n) for (int i = 1; i <= (n); i++)
#define PB push_back
#define _clr(x) memset(x, 0, sizeof(x))
#define SZ(x) ((int) (x).size())
typedef vector<int> VI;

const int N = 1e5 + 100;
//VI rel[N];
int bg[N], ed[N];
bool vis[N];
int ebg[N], enxt[N << 1], eadj[N << 1], edgeCnt;
int bit[N], bitSZ;

void init(int n) {
    _clr(bit);
    edgeCnt = 0;
    REP_1(i, n) {
        ebg[i] = enxt[i] = -1;
    }
}

void addedge(int u, int v) {
    eadj[edgeCnt] = v;
    enxt[edgeCnt] = ebg[u];
    ebg[u] = edgeCnt++;
}

void add2edge(int u, int v) {
    addedge(u, v);
    addedge(v, u);
}

void input(int n) {
    REP_1(i, n)
//        rel[i].clear(),
        vis[i] = false;
    n--;
    int x, y;
    REP(i, n) {
        scanf("%d%d", &x, &y);
//        rel[x].PB(y);
//        rel[y].PB(x);
        add2edge(x, y);
    }
}

void add(int x, int d) {
    for (; x <= bitSZ; x += (-x) & (x)) {
        bit[x] += d;
    }
}

int sum(int x) {
    int ret = 0;
    for (; x > 0; x -= (-x) & x) {
        ret += bit[x];
    }
    return ret;
}

int curID;

void dfs(int x) {
    add(curID, 1);
    vis[x] = true;
//    int sz = SZ(rel[x]);
    int t;
    bg[x] = curID++;
//    REP(i, sz) {
    for (int eid = ebg[x]; ~eid; eid = enxt[eid]) {
//        t = rel[x][i];
        t = eadj[eid];
        if (vis[t]) continue;
        dfs(t);
    }
    ed[x] = curID;
}

void makeList(int n) {
    curID = 1;
    dfs(1);
//    REP_1(i, n) {
//        cout << i << ": " << bg[i] << ' ' << ed[i] << endl;
//    }
}

int sum(int l, int r) {
    return sum(r) - sum(l);
}

int main() {
//    freopen("in", "r", stdin);
    int n, x;
    char buf[3];
    while (~scanf("%d", &n)) {
        bitSZ = n;
        init(n);
        input(n);
        makeList(n);
        scanf("%d", &n);
        REP(i, n) {
            scanf("%s%d", buf, &x);
            if (buf[0] == 'Q') {
//                cout << bg[x] - 1 << ' ' << ed[x] - 1 << endl;
                printf("%d\n", sum(bg[x] - 1, ed[x] - 1));
            } else {
//                cout << "~~ " << bg[x] << endl;
                if (vis[x]) {
                    add(bg[x], -1);
                    vis[x] = false;
                } else {
                    add(bg[x], 1);
                    vis[x] = true;
                }
            }
        }
    }
    return 0;
}

——written by Lyon