LCA及应用

最新推荐文章于 2024-06-27 10:04:18 发布

codeisking

最新推荐文章于 2024-06-27 10:04:18 发布

阅读量211

点赞数

分类专栏：算法文章标签：算法

本文链接：https://blog.youkuaiyun.com/codeisking/article/details/120623870

版权

算法专栏收录该内容

7 篇文章

订阅专栏

该代码实现了一个基于LCA（最近公共祖先）的算法，用于在一个带权重的树形图中查找两个节点间的最短路径。通过深度优先搜索遍历树并记录路径，然后计算给定起点和终点之间的距离。输入为树的节点数、边数及边的连接信息，输出为多个查询对之间的最短距离。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
#define N 1005
struct Node{int v,w;};
vector<Node> G[N];
int n,m,st[N][12],dep[N],d[N];
void DFS(int u,int p){
    st[u][0]=p;
    for(int j=1;j<12;j++)st[u][j]=st[st[u][j-1]][j-1];
    for(int i=0;i<G[u].size();i++)if(G[u][i].v!=p)
        dep[G[u][i].v]=dep[u]+1,d[G[u][i].v]=d[u]+G[u][i].w, DFS(G[u][i].v,u);
}
int LCA(int u,int v){
    if(dep[u]<dep[v])swap(u,v);
    for(int i=0,d=dep[u]-dep[v];d;i++,d>>=1)if(d&1)u=st[u][i];
    if(u==v)return u;
    for(int i=11;~i;i--)if(st[u][i]!=st[v][i])u=st[u][i],v=st[v][i];
    return st[u][0];
}
int dist(int u,int v){return d[u]+d[v]-2*d[LCA(u,v)];}
int main(){
    scanf("%d%d",&n,&m);
    for(int i=1,u,v,w;i<n;i++)scanf("%d%d%d",&u,&v,&w),G[u].push_back((Node){v,w}),G[v].push_back((Node){u,w});
    DFS(1,0);
    while(m--){
        int a,b;
        scanf("%d%d",&a,&b);
        printf("%d\n",dist(a,b));
    }
    return 0;
}