距离 (离线tarjan)

//tarjan
//d预处理每个点到根节点的距离,dx + dy 减去2倍的最近公共祖先到根节点的距离即可  == dx + dy - 2dp,所以就是问你最近公共祖先
    //做法3:离线做法:把所有询问全部都进来,然后再输出
/*targan---离线求lca, 可以做到O(n + m); 本质是对向上标记法的一个优化
    基于深度优先遍历, 在这个过程中将所有的点分成三大类     
    第一类:已经遍历过且已经回溯过的点              2
    第二类:正在搜索的分支(就是从root到一个节点的路径)          1   
    第三类:还未搜索到的点                              0
    当询问到第二类中的点的时候,我们可以发现第一类中的所有子树上的点其实都可以合并到它的根上去(并查集优化)
    
*/


#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>
#include <vector>

using namespace std;

typedef pair<int, int> PII;

const int N = 10010, M = N * 2;

int n, m;
int h[N], e[M], w[M], ne[M], idx;
int dist[N];
int p[N];
int res[M];
int st[N];
vector<PII> query[N];   // first存查询的另外一个点,second存查询编号

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

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

void dfs(int u, int fa)
{
    for (int i = h[u]; ~i; 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; i = ne[i])
    {
        int j = e[i];
        if (!st[j])//如果没搜过就搜
        {
            tarjan(j);
            p[j] = u;//搜完就合并
        }
    }
    
    for (auto item : query[u])//离线查询对节点u的询问
    {
        int y = item.first, id = item.second;
        if (st[y] == 2)//说明搜完了
        {
            int anc = find(y);
            res[id] = dist[u] + dist[y] - 2 * dist[anc];
        }
    }
    st[u] = 2;
}

int main()
{
    scanf("%d%d", &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 = 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});
        }
    }

    for (int i = 1; i <= n; i ++ )  p[i] = i;
    dfs(1, -1);//处理距离,u 和 fa
    tarjan(1);
    
    for (int i = 0; i < m; i ++ )cout << res[i] << endl;
    return 0;
}

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值