HDU - 2586 - How far away ？（倍增LCA）

最新推荐文章于 2022-07-23 01:36:13 发布

sugarbliss

最新推荐文章于 2022-07-23 01:36:13 发布

阅读量240

点赞数

分类专栏：【最近公共祖先LCA】

本文链接：https://blog.csdn.net/sugarbliss/article/details/92837598

版权

【最近公共祖先LCA】专栏收录该内容

3 篇文章 0 订阅

订阅专栏

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=2586

题意：给你一颗无根树，让你求树上 u,v 两点的距离。

思路：以任意一点为根节点建一颗树，然后求询问两点 u,v 的 LCA(u,v)=rt ，然后用 dis[x] 表示根节点到点的距离那么答案就是， ans = dis[x]-dis[rt]+dis[y]-dis[rt] 。

#include <bits/stdc++.h>
using namespace std;
const int N = 4e4 + 7;
int n, m, lg[N], deep[N], fa[N][20], dis[N];
#define pii pair <int, int>
vector <pii> E[N];
void dfs(int x, int fath)
{
    fa[x][0] = fath;
    deep[x] = deep[fath] + 1;
    for(int i = 1; (1 << i) <= n; i++)
        fa[x][i] = fa[fa[x][i-1]][i-1];
    for(int i = 0; i < E[x].size(); i++)
    {
        int v = E[x][i].first;
        if(v != fath)
        {
            dis[v] = dis[x] + E[x][i].second;
            dfs(v, x);
        }
    }
}
int lca(int x, int y)
{
    if(deep[y] > deep[x])
        swap(x, y);
    while(deep[x] > deep[y])
        x = fa[x][lg[deep[x]-deep[y]]];
    if(x == y) return x;
    for(int i = lg[deep[x]]; i >= 0; i--)
        if(fa[x][i] != fa[y][i])
            x = fa[x][i], y = fa[y][i];
    return fa[x][0];
}
int main()
{
    int T; scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n, &m);
        for(int i = 1; i < n; i++)
        {
            int u, v, w;
            scanf("%d%d%d",&u, &v, &w);
            E[u].push_back({v, w});
            E[v].push_back({u, w});
        }
        lg[0] = -1;
        for(int i = 1; i <= n; i++)
            lg[i] = lg[i >> 1] + 1;
        dfs(1, 0);
        while(m--)
        {
            int x, y, ans;
            scanf("%d%d",&x, &y);
            ans = dis[x] + dis[y] - 2 * dis[lca(x, y)];
            printf("%d\n",ans);
        }
    }
}