距离（AcWing）

最新推荐文章于 2025-08-01 19:44:23 发布

原创最新推荐文章于 2025-08-01 19:44:23 发布 · 127 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

算法题

该文章描述了一个关于图论的问题，给定一棵有n个点的树，需要处理m次询问，每次询问给出两点间在树上的最短距离。解决方案使用了路径压缩的最近公共祖先(LCA)算法和深度优先搜索(DFS)，以O(nlog^2n)的时间复杂度预处理，然后在线性时间内回答每个询问。

给出 n个点的一棵树，多次询问两点之间的最短距离。

注意：

边是无向的。
所有节点的编号是 1,2,…,n.

输入格式

第一行为两个整数 n和 m。n 表示点数，m 表示询问次数；

下来 n−1 行，每行三个整数 x,y,k，表示点 x 和点 y 之间存在一条边长度为 k；

再接下来 m 行，每行两个整数 x,y，表示询问点 x 到点 y 的最短距离。

树中结点编号从 1 到 n。

输出格式

共 m 行，对于每次询问，输出一行询问结果。

数据范围

$2\leqslant n\leqslant 10^{4}$
$1\leqslant m\leqslant 2*10^{4}$
$0\leqslant k\leqslant 1000$
$1\leqslant x,y\leqslant n$

思路

模板，LCA求树上两点路径长度，时间复杂度 $O(logn)$

#include<bits/stdc++.h>
using namespace std;
const int N=1e4+10;
int f[N][21];
int d[N];
int dep[N];
struct node{
    int v;
    int e;
};
vector<node> edge[N];
int n,m;
void dfs(int u,int fa){
    f[u][0]=fa;
    for(int i=1;i<=20;i++){
        f[u][i]=f[f[u][i-1]][i-1];
    }
    for(int i=0;i<edge[u].size();i++){
        int v=edge[u][i].v;
        int e=edge[u][i].e;
        if(v==fa) continue;
        d[v]=d[u]+e;
        dep[v]=dep[u]+1;
        dfs(v,u);
    }
    return ;
}
int lca(int x,int y){
    if(dep[x]>dep[y]) swap(x,y);
    int dd=dep[y]-dep[x];
    for(int i=0;i<=20;i++){
        if((1<<i)&dd) y=f[y][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 main(void){

    scanf("%d%d",&n,&m);
    for(int i=1;i<=n-1;i++){
        int x,y,k;
        scanf("%d%d%d",&x,&y,&k);
        edge[x].push_back(node{y,k});
        edge[y].push_back(node{x,k});
    }
    d[1]=0;
    dep[1]=0;
    dfs(1,0);
    for(int i=1;i<=m;i++){
        int x,y;
        scanf("%d%d",&x,&y);
        int u=lca(x,y);
        int ans;
        if(u==y){
            ans=d[x]-d[y];
        }
        else if(u==x){
            ans=d[y]-d[x];
        }
        else{
            ans=d[x]+d[y]-2*d[u];
        }
        printf("%d\n",ans);
    }


    return 0;
}

标签

#算法 #数据结构