题目链接:树上距离
分析
倍增求LCA,预处理 f a fa fa数组的时候顺便求根节点到所有点的距离dis。
然后
a
n
s
=
d
i
s
[
x
]
+
d
i
s
[
y
]
−
2
∗
d
i
s
[
l
c
a
(
x
,
y
)
]
ans=dis[x]+dis[y]-2*dis[lca(x,y)]
ans=dis[x]+dis[y]−2∗dis[lca(x,y)]
(x到根的距离+y到根的距离-两倍LCA到根的距离)
上代码
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
int n,m;
int tot,hd[50001];
int fa[50001][17],dep[50001],dis[50001];
struct node
{
int to,next,w;
}e[50001];
void add(int x,int y,int z)
{
e[++tot]=(node){y,hd[x],z};
hd[x]=tot;
}
void dfs(int x,int fx)
{
fa[x][0]=fx;
dep[x]=dep[fx]+1;
for(int i=1;i<=16;i++)
{
fa[x][i]=fa[fa[x][i-1]][i-1];
}
for(int i=hd[x];i>0;i=e[i].next)
{
if(e[i].to==fx) continue;
dis[e[i].to]=dis[x]+e[i].w;
dfs(e[i].to,x);
}
}
int lca(int x,int y)
{
if(dep[x]<dep[y]) swap(x,y);
for(int i=16;i>=0;i--)
{
if(dep[fa[x][i]]>=dep[y])
{
x=fa[x][i];
}
}
if(x==y) return x;
for(int i=16;i>=0;i--)
{
if(fa[x][i]!=fa[y][i])
{
x=fa[x][i];
y=fa[y][i];
}
}
return fa[x][0];
}
int main()
{
cin>>n>>m;
for(int i=1;i<=n-1;i++)
{
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
add(x,y,z);
add(y,x,z);
}
dfs(1,0);
for(int i=1;i<=m;i++)
{
int x,y;
scanf("%d%d",&x,&y);
cout<<dis[x]+dis[y]-2*dis[lca(x,y)]<<endl;
}
return 0;
}