洛谷 P4092 [HEOI2016/TJOI2016]树 || bzoj4551

https://www.lydsy.com/JudgeOnline/problem.php?id=4551

https://www.luogu.org/problemnew/show/P4092

这当然是树剖裸题，还可以不写线段树用set

（只用set达到一个log是不行的，询问中，中间遇到的路径并不总是整段的轻/重链，可能会有半段的）

 #include<cstdio>
 #include<algorithm>
 #include<cstring>
 #include<vector>
 #include<set>
 using namespace std;
 #define fi first
 #define se second
 #define mp make_pair
 #define pb push_back
 typedef long long ll;
 typedef unsigned long long ull;
 typedef pair<int,int> pii;
 struct E
 {
     int to,nxt;
 }e[];
 int f1[],ne;
 int n,q;
 void me(int a,int b)
 {
     e[++ne].to=b;e[ne].nxt=f1[a];f1[a]=ne;
 }
 int hson[],tp[],sz[],dp[];
 int ar[],lp[],f[];
 struct Cmp
 {
     bool operator()(int a,int b)
     {
         return dp[a]<dp[b];
     }
 };
 set<int,Cmp> t2[];//以此点为top的链上所有标记点
 void dfs1(int u,int fa)
 {
     sz[u]=;
     for(int k=f1[u];k;k=e[k].nxt)
         if(e[k].to!=fa)
         {
             dp[e[k].to]=dp[u]+;
             dfs1(e[k].to,u);
             sz[u]+=sz[e[k].to];
             if(sz[e[k].to]>sz[hson[u]])
                 hson[u]=e[k].to;
         }
 }
 void dfs2(int u,int fa)
 {
     ar[++ar[]]=u;lp[u]=ar[];
     f[u]=fa;
     if(u==hson[fa]) tp[u]=tp[fa];
     else    tp[u]=u;
     if(hson[u]) dfs2(hson[u],u);
     for(int k=f1[u];k;k=e[k].nxt)
         if(e[k].to!=fa&&e[k].to!=hson[u])
             dfs2(e[k].to,u);
 }
 int main()
 {
     int i,a,b;
     char tmp[];
     scanf("%d%d",&n,&q);
     for(i=;i<n;i++)
     {
         scanf("%d%d",&a,&b);
         me(a,b);me(b,a);
     }
     dfs1(,);dfs2(,);
     //for(i=1;i<=n;i++)    printf("%d\n",f[i]);
     //puts("b");
     //for(i=1;i<=n;i++)    printf("%d\n",ar[i]);
     //puts("c");
     t2[].insert();//tag[1]=1;
     while(q--)
     {
         scanf("%s",tmp);
         if(tmp[]=='C')
         {
             scanf("%d",&a);
             t2[tp[a]].insert(a);
         }
         else
         {
             scanf("%d",&a);
             set<int,Cmp>::iterator it;
             while()
             {
                 it=t2[tp[a]].upper_bound(a);
                 if(it!=t2[tp[a]].begin())
                 {
                     --it;
                     printf("%d\n",*it);
                     break;
                 }
                 a=f[tp[a]];
             }
         }
     }
     return ;
 }