1171. 距离

给出 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≤n≤104,
1≤m≤2×104,
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

思路：

/*
Tarjan – 离线求LCA
在线做法：读一个询问，处理一个，输出一个（倍增法）
离线做法：读完全部询问，再全部处理完，再全部输出

在深度优先遍历时，将所有点分成三大类
2号点：代表已经访问并结束回溯(重点)
1号点：代表正在访问
0号点：代表还没有访问过
其中所有2号点和正在搜索的1号点路径中已经通过并查集合并成一个集合

1、先求出所有点到根结点的距离depth[]，设x号点和y号点的最近公共祖先是p,则x和y的最近距离等于depth[x] + depth[y] - 2 * depth[p]
2、在深度优先遍历1号点中的u点的时候，需要把u的查询的另外一个点的最短距离进行计算并存储，最后把u点合并到上一结点的集合
*/

代码：

#include <bits/stdc++.h>
using namespace std;
typedef pair<int, int> PII;

const int N = 1e4 + 10, M = N * 2;
int h[N], ne[M], e[M], w[M], idx;
int n, m;
int st[N], p[N], dist[N], res[M];
vector<PII> query[N];

void add(int a, int b, int c)
{
    e[idx] = b;
    w[idx] = c;
    ne[idx] = h[a];
    h[a] = idx++;
}

int find(int x)
{
    if (p[x] != x)
        p[x] = find(p[x]);
    return p[x];
}

void dfs(int u, int fa) //算距离的技巧
{
    for (int i = h[u]; i != -1; i = ne[i])
    {
        int j = e[i];
        if (j == fa)
            continue;
        dist[j] = dist[u] + w[i];
        dfs(j, u);
    }
}

void tarjan(int u)
{
    st[u] = 1;
    for (int i = h[u]; i != -1; i = ne[i])
    {
        int j = e[i];
        if (!st[j])
        {
            tarjan(j);
            p[j] = u;
        }
    }

    for (auto it : query[u]) // 处理已经访问过的点和当前点的最长距离
    {
        int y = it.first, id = it.second;
        if (st[y] == 2)
        {
            int ans = find(y);
            res[id] = dist[y] + dist[u] - 2 * dist[ans];
        }
    }

    st[u] = 2;
}

int main()
{
    cin >> n >> m;
    memset(h, -1, sizeof(h));
    for (int i = 0; i < n - 1; i++)
    {
        int a, b, c;
        scanf("%d %d %d", &a, &b, &c);
        add(a, b, c), add(b, a, c);
    }

    for (int i = 1; i <= n; i++)
        p[i] = i;

    for (int i = 0; i < m; i++)
    {
        int a, b;
        scanf("%d%d", &a, &b);
        if (a != b)
        {
            query[a].push_back({b, i});
            query[b].push_back({a, i});
        }
    }

    dfs(1, -1);

    tarjan(1);

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

    return 0;
}