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

题面：BZOJ4551
这题在线最优复杂度O(nlogn)，离线则是O(nα)，具体做法网上都有
然而我有一个神奇的O(nlog2n)的想法
其实和在线线段树差不多
如果有一个点被标记，我们把这棵子树的权值全部+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;}