51nod-2621 树上距离（tarjian离线求LCA）

51nod-2621 树上距离

题目

求树上两点最短距离，q次查询

分析

按有向树存，选出一个根节点。

首先预处理节点到根节点的距离 $dis$，用 tarjian离线求询问两点的LCA。

最后套结果 $dis(u,v)=dis[u]+dis[v]-2\ast dis[LCA]$

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
const ll mod = 9999991;
const int N = 1e3 + 10;

int n, q, nn, x, y, z, r;
struct node{
    int v, l;                       //存树时，v 代表指向的节点，l 边距离
    node(int v, int l):v(v),l(l){}  //存查询时，v 代表关系节点，l 查询序号
};
vector<node> tr[N];     //有向树
vector<node> ask[N];     //查询
int rt[N], dis[N], fa[N], vis[N], ans[N];

int find(int x){            //并查集
    return fa[x] = fa[x] == x ? x : find(fa[x]);
}

void tarjian(int rt){                   //就是 dfs 序遍历树
    for(auto x : tr[rt]){
        // d("fds");
        dis[x.v] = dis[rt] + x.l;     //求节点到根节点距离
        tarjian(x.v);
        fa[x.v] = rt;                   //回溯的时候标记
    }
    vis[rt] = 1;
    for(auto x : ask[rt]){
        if(vis[x.v]){                   //如果跟 rt 节点有关系的节点 x 访问过，LCA就是 find(x)
            // d("fds");
            ans[x.l] = dis[x.v] + dis[rt] - 2 * dis[find(x.v)];
        }
    }
}

int main()
{
    scanf("%d%d", &n, &q);
    for (int i = 0; i < n - 1; i++){
        scanf("%d%d%d", &x, &y, &z);
        tr[x].push_back(node(y, z));
        rt[y]++;
    }
    for(int i = 1; i <= n; i++){    //找跟节点
        fa[i] = i;
        if(!rt[i]){
            r = i;
        }
    }
    for (int i = 0; i < q; i++){
        scanf("%d%d", &x, &y);
        ask[x].push_back(node(y, i));
        ask[y].push_back(node(x, i));
    }
    tarjian(r);
    for(int i = 0; i < q; i++){
        printf("%d\n", ans[i]);
    }
    return 0;
}