http://wikioi.com/problem/1036/
由于n-1边连接n个点,所以不存在路径选择的问题,只能有一条路
而找2个点的距离就找到公共祖先的距离之和
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstdlib>
#include<cmath>
#include<cstring>
using namespace std;
int n,m;
struct edge
{
int v,next,w;
}e[30000*2+10];
int k=1;
int head[30000+10];
void add(int u,int v)
{
e[k].v=v;
e[k].w=1;
e[k].next=head[u];
head[u]=k++;
}
int vis[30000+10]={0};
int dep[30000+10];
int fa[30000+10]={0};
int s=0;
void dfs(int f,int x)//fa节点,x当前节点
{
if(vis[x])return;
dep[x]=dep[f]+1;
s=max(dep[x],s);
vis[x]=1;
for(int i=head[x];i!=-1;i=e[i].next)
{
int v=e[i].v;
if(!vis[v])
{
fa[v]=x;
dfs(x,v);
}
}
}//树的形态是自己确定的
int lca(int u,int v)
{
int nu=u,nv=v;
if(dep[v]<dep[u])swap(u,v);//深度不同要走到相同深度
if(dep[v]>dep[u])
{
while(dep[v]>dep[u])//这个不是正规的,应该多开一维记录第2^i个祖先
{
v=fa[v];
}
}
while(u!=v)//同时走
{
u=fa[u];
v=fa[v];
}
int final=v;
return dep[nu]+dep[nv]-2*dep[final];
}
int main()
{
memset(head,-1,sizeof(head));
cin>>n;
for(int i=1;i<n;i++)
{
int a,b;
scanf("%d%d",&a,&b);
add(a,b);
add(b,a);
}
dep[0]=0;
fa[1]=1;
dfs(0,1);//建树
cin>>m;
int last=1;
int ans=0;
for(int i=1;i<=m;i++)
{
int a;
scanf("%d",&a);
ans+=lca(last,a);
last=a;
}
cout<<ans<<endl;
}