[POJ](3321)Apple Tree ---- 樹形結構轉線性區間+樹狀數組

題目傳送門

做法：

• 這個題可以作爲樹狀數組的好題了~
• 難點不在於樹狀數組，而在於轉換問題。
• 數據結構不熟，看了其他人的博客，才知道可以通過先序遍歷這棵樹，進行轉換。
• 比如我現在建立一棵樹：
• 1 ---- 2
• 1 ---- 3
• 2 ---- 4
• 2 ---- 5
• 4 ---- 6
• 3 ---- 7
• 3 ---- 8
• 然後從根節點1，先序遍歷
• 次序爲：1 2 4 6 5 3 7 8
• 我們每訪問一個爲訪問過的結點就記錄一下，這是當前第幾個訪問，記錄起點和重點
• 那麼
• 節點1 ---> [1,8]
• 節點2 ---> [2,5]
• 節點4 ---> [3,4]
• 節點6 ---> [4,4]
• 節點5 ---> [5,5]
• 節點3 ---> [6,8]
• 節點7 ---> [7,7]
• 節點8 ---->[8,8]
• 醬紫我們就給每個結點分配了一個區間，我們用樹狀數組維護這些區間即可！

參考博客

 AC代碼：

#include <iostream>
#include <cmath>
#include <cstdio>
#include <algorithm>
#include <cstring>
#include <vector>
#include <ctime>
#define IO          ios_base::sync_with_stdio(0),cin.tie(0),cout.tie(0)
#define pb(x)       push_back(x)
#define sz(x)       (int)(x).size()
#define sc(x)       scanf("%d",&x)
#define pr(x)       printf("%d\n",x)
#define abs(x)      ((x)<0?-(x):x)
#define all(x)      x.begin(),x.end()
using namespace std;
typedef long long ll;
const int mod = 1e9+7;
const double PI = 4*atan(1.0);
const int maxm = 1e8+5;
const int maxn = 1e5+5;
const int INF = 0x3f3f3f3f;
int n,m,step;
int c[maxn];
int apple[maxn];
bool vis[maxn];
vector<vector<int> >vec(maxn);
struct node
{
    int l,r;
}line[maxn];
inline int lowbit(int x){return x&(-x);}
void update(int x,int val)
{
    for(int i=x;i<=n;i+=lowbit(i)) c[i]+=val;
}
int query(int x)
{
    int res = 0;
    for(int i=x;i;i-=lowbit(i)) res+=c[i];
    return res;
}
void init()
{
    memset(apple,0,sizeof(apple));
    memset(vis,false,sizeof(vis));
    memset(c,0,sizeof(c));
    step = 0;
    for(int i=0;i<maxn;i++) vec[i].clear();
}
void dfs(int x) //先序遍歷
{
    vis[x] = true;
    line[x].l = ++step;
    for(int i=0;i<sz(vec[x]);i++)
    {
        if(vis[vec[x][i]]) continue;
        dfs(vec[x][i]);
    }
    line[x].r = step;
}
int main()
{
    #ifdef LOCAL_FILE
    freopen("in.txt","r",stdin);
    #endif // LOCAL
    while(~scanf("%d",&n))
    {
        init();
        for(int i=1;i<=n-1;i++)
        {
            int u,v;
            sc(u);sc(v);
            vec[u].pb(v);
            vec[v].pb(u);
        }
        dfs(1);
        sc(m);
        for(int i=1;i<=n;i++)
        {
            update(i,1); //區間
            apple[i] = 1;
        }
        for(int i=0;i<m;i++)
        {
            int x;
            char op[2];
            scanf("%s %d",op,&x);
            if(op[0] == 'C'){  //維護的區間是每個結點所對應的區間
                int pos = line[x].l;
                if(apple[pos] == 0) update(pos,1);
                if(apple[pos] == 1) update(pos,-1);
                apple[pos] = !apple[pos];
            }
            else if(op[0] == 'Q'){
                int ans;
                int l = line[x].l,r = line[x].r;
                if(l == 1) ans = query(r);
                else ans = query(r)-query(l-1);
                printf("%d\n",ans);
            }
        }
    }
    #ifdef LOCAL_FILE
    cout << "Time elapsed: " << 1.0 * clock() / CLOCKS_PER_SEC * 1000 << "ms." << endl;
    #endif // LOCAL_FILE
    return 0;
}