【CODE[VS]】1228 蘋果樹 樹狀數組

題目傳送門

爲什麼我拿到這題的第一反應是樹剖啊……我到底在想些什麼啊

然而這題被分類在樹狀數組裏，於是我就把想法往樹狀數組上靠，依然沒有任何思路

於是我放棄了樹狀數組的解法，想用樹剖暴艹這題。

卻突然驚訝的發現：樹剖你妹啊，這題不就是先序遍歷後樹狀數組維護節點權值嗎？

被一道傻逼題浪費了一個小時的我表示有一句mmp不知當不當講……

附上AC代碼：

#include <cstdio>
#include <cctype>
using namespace std;

const int N=1e5+10;
struct side{
    int to,nt;
}s[N<<1];
int n,h[N],num,wz[N],sz[N],size,q,a[N],t[N],x,y;

inline char nc(void){
    static char ch[100010],*p1=ch,*p2=ch;
    return p1==p2&&(p2=(p1=ch)+fread(ch,1,100010,stdin),p1==p2)?EOF:*p1++;
}

inline void read(int &a){
    static char c=nc();int f=1;
    for (;!isdigit(c);c=nc()) if (c=='-') f=-1;
    for (a=0;isdigit(c);a=(a<<3)+(a<<1)+c-'0',c=nc());
    return (void)(a*=f);
}

inline void add(int x,int y){
    s[++num]=(side){y,h[x]},h[x]=num;
    s[++num]=(side){x,h[y]},h[y]=num;
}

inline void so(int x,int fa){
    wz[x]=++size,sz[x]=1;
    for (int i=h[x]; i; i=s[i].nt)
        if (s[i].to!=fa) so(s[i].to,x),sz[x]+=sz[s[i].to];
    return;
}

#define lowbit(x) ((x)&(-x))
inline void updata(int x,int w){for (int i=x; i<=n; i+=lowbit(i)) t[i]+=w; return;}
inline int query(int x){int ret=0; for (int i=x; i; i-=lowbit(i)) ret+=t[i]; return ret;}

int main(void){
    read(n);
    for (int i=1; i<n; ++i) read(x),read(y),add(x,y);
    so(1,0),read(q);
    for (int i=1; i<=n; ++i) a[i]=1,updata(wz[i],1);
    while (q--){
        char c=nc();while (c!='C'&&c!='Q') c=nc();read(x);
        if (c=='C')
            if (a[x]) a[x]=0,updata(wz[x],-1);
            else a[x]=1,updata(wz[x],1);
        else printf("%d\n",query(wz[x]+sz[x]-1)-query(wz[x]-1));
    }
    return 0;
}