【codevs 1036】商务旅行

lca

#include<cstdio>
#include<iostream>
#include<algorithm>
#include<queue>
#include<cmath>
#include<cstring>
using namespace std;
const int maxn=30000+5;
int fa[maxn][25],deep[maxn],dist[maxn];
int n,m,tot,first[maxn],next[maxn<<1];
struct edge
{
    int f,t,v;
}es[maxn<<1];
void build(int f,int t,int v)
{
    es[++tot]=(edge){f,t,v};
    next[tot]=first[f];
    first[f]=tot;
}
void init(int q,int h)
{
    for(int i=first[h];i;i=next[i])
    {
        int at=es[i].t,av=es[i].v;
        if(at==q) continue;
        fa[at][0]=h;
        dist[at]=dist[h]+av;
        deep[at]=deep[h]+1;
        init(h,at);
    }
}
void make_lca()
{
    for(int i=1;i<=log2(n);i++)
      for(int j=1;j<=n;j++)
        fa[j][i]=fa[fa[j][i-1]][i-1];
}
int lca(int x,int y)
{
    if(deep[x]<deep[y]) swap(x,y);
    for(int i=log2(m);i>=0;i--)
    {
        if(deep[fa[x][i]]>=deep[y])
            x=fa[x][i];
    }
    if(x==y) return x;
    for(int i=log2(n);i>=0;i--)
    {
        if(fa[x][i]!=fa[y][i])
        {
            x=fa[x][i];
            y=fa[y][i];
        }
    }
    return fa[x][0];
}
int main()
{
    scanf("%d",&n);
    for(int i=1;i<n;i++)
    {
        int v,u;
        scanf("%d%d",&v,&u);
        build(v,u,1);
        build(u,v,1);
    }
    init(1,1);
    make_lca();
    scanf("%d",&m);
    int ans=0;
    for(int i=1;i<=m;i++)
    {
        int a,la;
        scanf("%d",&a);
        if(i==1)  la=a;
        else 
        {
            ans+=dist[a]+dist[la]-2*(dist[lca(a,la)]);
            la=a;
        }
    }
    printf("%d\n",ans);
    return 0;
}