Qtree3 - 树链剖分

打完以后才发现写复杂了……算了懒得改了

 #include <bits/stdc++.h>
 using namespace std;

 int dep[],fa[][],size[],wson[],vis[],sid[],tid[];
 int n,m,t1,t2,t3,t4,ind,top[],a[];

 vector <int> g[];

 void dfs1(int p) {
     size[p]=;
     vis[p]=;
     for(int i=;i<g[p].size();i++) {
         if(vis[g[p][i]]) continue;
         dep[g[p][i]]=dep[p]+;
         fa[g[p][i]][]=p;
         dfs1(g[p][i]);
         size[p]+=size[g[p][i]];
         if(size[g[p][i]]>size[wson[p]]) wson[p]=g[p][i];
     }
 }

 void dfs2(int p) {
     vis[p]=;
     sid[p]=++ind;
     tid[ind]=p;
     if(wson[p]) {
         top[wson[p]]=top[p];
         dfs2(wson[p]);
     }
     for(int i=;i<g[p].size();i++) {
         if(vis[g[p][i]]) continue;
         top[g[p][i]]=g[p][i];
         dfs2(g[p][i]);
     }
 }

 void lca_presolve() {
     for(int i=;i<=;i++)
         for(int j=;j<=n;j++)
             fa[j][i]=fa[fa[j][i-]][i-];
 }

 int lca(int p,int q) {
     if(dep[p]<dep[q]) swap(p,q);
     for(int i=;i>=;i--) if(dep[fa[p][i]]>=dep[q]) p=fa[p][i];
     for(int i=;i>=;i--) if(fa[p][i]-fa[q][i]) p=fa[p][i], q=fa[q][i];
     if(p-q) p=fa[p][], q=fa[q][];
     return max(p,q);
 }

 void pushup(int p){
     if(a[p*]) a[p]=a[p*];
     else a[p]=a[p*+];
 }

 void modify(int p,int l,int r,int pos) {
     if(l==r) {
         a[p]=a[p]?:l;
         return;
     }
     if(pos<=(l+r)/) modify(p*,l,(l+r)/,pos);
     else modify(p*+,(l+r)/+,r,pos);
     pushup(p);
 }

 int query(int p,int l,int r,int ql,int qr) {
     if(l>qr||r<ql) return ;
     if(l>=ql&&r<=qr) return a[p];
     int q0=query(p*,l,(l+r)/,ql,qr);
     if(q0) return q0;
     else return query(p*+,(l+r)/+,r,ql,qr);
 }

 void tmodify(int pos) {
     modify(,,n,sid[pos]);
 }

 int lquery_upward(int anc,int son) {
     int ans=;
     while(dep[top[son]]>dep[anc]) {
         ans=query(,,n,sid[top[son]],sid[son]);
         son=fa[top[son]][];
         if(ans) return ans;
     }
     return query(,,n,sid[anc],sid[son]);
 }

 int lquery_downward(int anc,int son) {
     int ans=,tmp;
     while(dep[top[son]]>dep[anc]) {
         tmp=query(,,n,sid[top[son]],sid[son]);
         if(tmp) ans=tmp;
         son=fa[top[son]][];
     }
     tmp=query(,,n,sid[anc],sid[son]);
     if(tmp) ans=tmp;
     return ans;
 }

 int tquery(int p,int q){
     int l=lca(p,q),tmp;
     tmp=lquery_upward(l,p);
     if(tmp) return tmp;
     else return lquery_downward(l,q);
 }

 int main(){
     ios::sync_with_stdio(false);
     cin>>n>>m;
     for(int i=;i<n;i++) cin>>t1>>t2, g[t1].push_back(t2), g[t2].push_back(t1);
     dep[]=; dfs1();
     memset(vis,,sizeof vis);
     dfs2();
     lca_presolve();

     for(int i=;i<=m;i++) {
         cin>>t1>>t2;
         if(t1==) tmodify(t2);
         else t3=tid[tquery(,t2)], cout<<(t3==?-:t3)<<endl;
     }
 }