【最近公共祖先】【块状树】CODEVS 1036 商务旅行

在线块状树LCA模板。

 #include<cstdio>
 #include<vector>
 #include<algorithm>
 #include<cmath>
 using namespace std;
 #define N 30001
 vector<int>G[N];
 typedef vector<int>::iterator ITER;
 int dep[N],x,y,fa[N],top[N],siz[N],sz,n,m,ans;
 int Abs(const int &x){return x<?(-x):x;}
 void makeblock(int U)
 {
     for(ITER it=G[U].begin();it!=G[U].end();it++)
       if((*it)!=fa[U])
         {
           fa[*it]=U;
           dep[*it]=dep[U]+;
           if(siz[top[U]]<sz)
             {
               top[*it]=top[U];
               siz[top[U]]++;
             }
           makeblock(*it);
         }
 }
 int lca(int U,int V)
 {
     while(U!=V)
       {
         if(top[U]!=top[V])
           {
             if(dep[top[U]]<dep[top[V]]) swap(U,V);
             U=fa[top[U]];
           }
         else
           {
             if(dep[U]<dep[V]) swap(U,V);
             U=fa[U];
           }
       }
     return U;
 }
 int main()
 {
     scanf("%d",&n);
     for(int i=;i<n;i++)
       {
         scanf("%d%d",&x,&y);
         G[x].push_back(y);
         G[y].push_back(x);
       } sz=(int)sqrt((double)n);
     for(int i=;i<=n;i++) siz[i]=,top[i]=i;
     makeblock(); scanf("%d%d",&m,&x);
     for(int i=;i<=m;i++)
       {
         scanf("%d",&y); int f=lca(x,y);
         ans+=(dep[x]+dep[y]-(dep[f]<<));
         x=y;
       }
     printf("%d\n",ans);
     return ;
 }