倍增+树状数组——BZOJ4551 [Tjoi2016&Heoi2016]树

题面：BZOJ4551
这题在线最优复杂度$O(nlogn)$，离线则是$O(nα)$，具体做法网上都有
然而我有一个神奇的$O(nlog^2n)$的想法
其实和在线线段树差不多
如果有一个点被标记，我们把这棵子树的权值全部+1
然后查询的时候向上找到最上面的权值和这个点相同的点
倍增一下就好了
我大概是石乐志了QAQ

#include <cstdio>
#include <algorithm>
#include <cmath>
#include <cstring>
#include <iostream>
#include <ctime>
#include <map>
#include <queue>
#include <cstdlib>
#include <string>
#include <climits>
#include <set>
#include <vector>
using namespace std;
inline int read(){
    int k=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){k=k*10+ch-'0';ch=getchar();}
    return k*f;
}
inline void write(int x){
    if(x<0)putchar('-'),x=-x;
    if(x>9)write(x/10);putchar(x%10+'0');
}
inline void writeln(int x){
    write(x);puts("");
}
bool vis[100010];
int nedge=0,p[200010],nex[200010],head[200010];
int n,q,f[100010][18],sx[100010],cnt=0,s[100010];
int sz[100010];
inline void addedge(int a,int b){
    p[++nedge]=b;nex[nedge]=head[a];head[a]=nedge;
}
inline void add(int x,int v){for(;x<=n;x+=x&-x)sz[x]+=v;}
inline int ssum(int x){int sum=0;for(;x;x-=x&-x)sum+=sz[x];return sum;}
inline void dfs(int x,int fa){
    sx[x]=++cnt;s[x]=1;f[x][0]=fa;
    for(int k=head[x];k;k=nex[k])if(p[k]!=fa){
        dfs(p[k],x);s[x]+=s[p[k]];
    }
}
inline int up(int x){
    int p=ssum(sx[x]);
    for(int i=17;i>=0;i--)if(ssum(sx[f[x][i]])==p)x=f[x][i];
    return x;
}
int main()
{
    n=read();q=read();
    for(int i=1;i<n;i++){
        int x=read(),y=read();
        addedge(x,y);addedge(y,x);
    }
    dfs(1,0);
    for(int j=1;j<=17;j++)
        for(int i=1;i<=n;i++)f[i][j]=f[f[i][j-1]][j-1];
    add(1,1);vis[1]=1;
    for(int i=1;i<=q;i++){
        char op[5];int x;scanf("%s",op);
        if(op[0]=='C'){
            x=read();
            if(!vis[x])vis[x]=1,add(sx[x],1),add(sx[x]+s[x],-1);
        }else x=read(),writeln(up(x));
    }
    return 0;
}