【lca+tarjan】距离

原题连接：https://www.acwing.com/problem/content/description/1173/

以前的倍增法求lca博客：https://blog.csdn.net/qq_44661312/article/details/97947398

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

注意：

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

输入格式

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

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

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

树中结点编号从 1 到 n。

输出格式

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

数据范围

2≤n≤1e4,
1≤m≤2e4,
0<k≤100,
1≤x,y≤n

输入样例1：

2 2 
1 2 100 
1 2 
2 1

输出样例1：

100
100

输入样例2：

3 2
1 2 10
3 1 15
1 2
3 2

输出样例2：

10
25

思路：把所有点分为三类，用st标记。其中，st==0代表还没遍历到的点， st==1代表正在处理的点，st==2代表已经处理好的点。

 

代码:

#include<stdio.h>
#include<vector>
#include<string.h>
#include<iostream>
using namespace std;
const int N = 1e4+10,M = 2e4+10;
struct query{
    int point,id;
    //id表示第i个查询，point表示第i个查询的另外一个顶点
};
int n,m;
int first[N],v[M],w[M],ne[M],num_edge;
int res[M];//存最后的答案
int dist[N],f[N],st[N];
//dist表示到根节点的距离，f表示父亲节点，st表示每个点的状态
vector<query>q[M];//记录每个询问

void add(int x,int y,int z){
    ne[++num_edge]=first[x];
    first[x]=num_edge;
    v[num_edge]=y;
    w[num_edge]=z;
}

void dfs(int x,int fa){
    int y;
    for(int i = first[x]; i != -1; i = ne[i]){
        y=v[i];
        if(y==fa)continue;
        dist[y]=dist[x]+w[i];
        dfs(y,x);
    }
}

int father(int x){//找祖宗
    if(x==f[x])return x;
    else return f[x]=father(f[x]);
}

void tarjan(int x){
    st[x]=1;//表示正在处理
    int y,num;
    //找出所有的子孙，并给予亲属关系
    //同时处理好所有子孙的关系，以便下面的计算
    for(int i = first[x]; i != -1; i = ne[i]){
        y=v[i];
        if(!st[y]){
            tarjan(y);
            f[y]=x;
            //这两句话不能交换顺序，否则可能会影响下方的father函数的结果
        }
    }

    num=q[x].size();
    for(int i = 0; i < num; i++){
        y=q[x][i].point;
        if(st[y]==2)//保证另一个点都已经处理好，确保答案的正确性
            res[q[x][i].id]=dist[x]+dist[y]-2*dist[father(y)];
            //与倍增法相似的方法，其中father（y）为两者最近公共祖先
    }
    st[x]=2;//表示已经处理好
}

int main(){
    scanf("%d%d",&n,&m);
    int a,b,c;
    memset(first,-1,sizeof(first));
    //建树
    for(int i = 1; i < n; i++){
        scanf("%d%d%d",&a,&b,&c);
        add(a,b,c),add(b,a,c);
    }
    //存下所有查询
    for(int i = 1; i <= m; i++){
        scanf("%d%d",&a,&b);
        q[a].push_back({b,i});
        q[b].push_back({a,i});
    }
    for(int i = 1; i <= n; i++)f[i]=i;

    dfs(1,-1);//预处理得到所有点到根节点的距离，这里设根节点为1
    tarjan(1);//算距离

    for(int i = 1; i <= m; i ++)printf("%d\n",res[i]);

}