专栏文章

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 条评论,欢迎与作者交流。

正在加载评论...