【最近公共祖先】【块状树】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<0?(-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]+1;
          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=1;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=1;i<=n;i++) siz[i]=1,top[i]=i;
    makeblock(1); scanf("%d%d",&m,&x);
    for(int i=2;i<=m;i++)
      {
        scanf("%d",&y); int f=lca(x,y);
        ans+=(dep[x]+dep[y]-(dep[f]<<1));
        x=y;
      }
    printf("%d\n",ans);
    return 0;
}