专栏文章
LCA
算法·理论参与者 1已保存评论 0
文章操作
快速查看文章及其快照的属性,并进行相关操作。
- 当前评论
- 0 条
- 当前快照
- 1 份
- 快照标识符
- @minjnsza
- 此快照首次捕获于
- 2025/12/02 03:30 3 个月前
- 此快照最后确认于
- 2025/12/02 03:30 3 个月前
CPP
#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+10;
int f[maxn][21],dep[maxn];
int head[maxn];
struct node
{
int w,nxt,to;
}e[maxn*2];
int w[maxn];
int n,m,tot;
void add(int u,int v,int z)
{
e[++tot].to=v;
e[tot].nxt=head[u];
e[tot].w=z;
head[u]=tot;
}
void dfs(int u,int fa)
{
dep[u]=dep[fa]+1;
for(int i=0;i<=19;++i) f[u][i+1]=f[f[u][i]][i];
for(int i=head[u];i;i=e[i].nxt)
{
int v=e[i].to;
if(v==fa) continue;
w[v]=w[u]+e[i].w;
f[v][0]=u;
dfs(v,u);
}
}
int lca(int x,int y)
{
if(dep[x]<dep[y]) swap(x,y);
for(int i=20;i>=0;--i)
{
if(dep[y]<=dep[f[x][i]]) x=f[x][i];
if(x==y) return x;
}
for(int i=20;i>=0;--i)
if(f[x][i]!=f[y][i]) x=f[x][i],y=f[y][i];
return f[x][0];
}
int dist(int x,int y)
{
return w[x]+w[y]-2*w[lca(x,y)];
}
int main()
{
scanf("%d%d",&n,&m);
for(int i=1;i<n;++i)
{
int x,y,z;
scanf("%d%d%d",&x,&y,&z);
add(x,y,z);add(y,x,z);
}
dfs(1,0);
while(m--)
{
int x,y;
scanf("%d%d",&x,&y);
printf("%d\n",dist(x,y));
}
return 0;
}
相关推荐
评论
共 0 条评论,欢迎与作者交流。
正在加载评论...