距离(AcWing)

给出 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
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 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;idep[y]) swap(x,y);
    int dd=dep[y]-dep[x];
    for(int i=0;i<=20;i++){
        if((1<=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;
}

你可能感兴趣的:(算法题,算法,数据结构)