目录:
题目:
分析:
题目跟我们说要求的是两点间的最短距离,我当然可以用dfs,但是,为了寻求更为高效的算法,我们采用了LCA算法
但有人就会问了,LCA不是用来求两点的最近公共祖先吗?当然,你没有记错,但我们只需一个公式就一样可以推出最短距离:
这个就是我们
的最短距离,dep为点的深度,而LCA则是用来求他们公共祖先是哪个点
思路:
1.建出邻接表
2.深搜求深度
3.常规LCA(树上倍增)求解
代码:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#define LL long long
using namespace std;
inline LL read() {
LL d=0,f=1;char s=getchar();
while(s<'0'||s>'9'){if(s=='-')f=-1;s=getchar();}
while(s>='0'&&s<='9'){d=d*10+s-'0';s=getchar();}
return d*f;
}
struct edge{
int to,next;
}e[60001];
int ls[60001],f[60001][17],dep[60001];
void dfs(int fa,int now)
{
f[now][0]=fa;
dep[now]=dep[fa]+1;
for(int i=ls[now];i;i=e[i].next)
if(!dep[e[i].to]) dfs(now,e[i].to);
}
int LCA(int l,int r)
{
if(dep[l]<dep[r]) swap(l,r);
for(int i=16;i>=0;i--)
if(dep[r]<=dep[f[l][i]])
l=f[l][i];
if(l==r) return l;
for(int i=16;i>=0;i--)
if(f[l][i]!=f[r][i])
{
l=f[l][i];
r=f[r][i];
}
return f[l][0];
}
int main()
{
int n=read(),l=0;
int a,b;
for(int i=2;i<=n;i++)
{
a=read();b=read();
e[++l]=(edge){b,ls[a]};
ls[a]=l;
e[++l]=(edge){a,ls[b]};
ls[b]=l;
}
dfs(0,1);
for(int j=1;j<=16;j++)
for(int i=1;i<=n;i++)
f[i][j]=f[f[i][j-1]][j-1];
//因为我们有m个点,所以需要滚动求解,还要累加
int now,m=read(),last=read(),ans=0;
for(int i=2;i<=m;i++)
{
now=read();
ans+=dep[last]+dep[now]-2*dep[LCA(last,now)];
last=now;
}
printf("%d",ans);
return 0;
}