How far away ? HDU - 2586 LCA (树上倍增详解)

9 篇文章 0 订阅

a,b两点间的距离为 dis[a]+dis[b]-2*dis[lca(a,b)] ,dis为到根节点的距离

#include<cstdio>
#include<cstring>
#include<cmath>
#include<queue>
using namespace std;
const int maxn=40010;
struct node
{
    int to,w;
    node(int a,int b)  //构造函数
    {
        to=a;
        w=b;
    }
};
vector<node>g[maxn];
int fa[maxn][17];  //表示节点i往上跳2^j次后的节点  注意j的取值范围
int d[maxn];    //当前节点所在的深度
int n,m;
int dis[maxn];  //到根节点的距离
bool vis[maxn];
int root;   //根结点
void init()  //初始化
{
    memset(fa,0,sizeof(fa));
    memset(vis,0,sizeof(vis));
    memset(d,0,sizeof(d));
    memset(dis,0,sizeof(dis));
}
void dfs(int u)
{
    int i;
    vis[u]=1;
    for(i=0;i<g[u].size();i++)  //每一个与u相连的节点i
    {
        int v=g[u][i].to;
        if(!vis[v])
        {
            d[v]=d[u]+1;  
            dis[v]=dis[u]+g[u][i].w;   //更新到根节点的距离
            dfs(v);
        }
    }
}//深搜出各点的深度,存在d中
void bz()
{
    int i,j;
    for(j=1;j<=16;j++)
        for(i=1;i<=n;i++)
        fa[i][j]=fa[fa[i][j-1]][j-1];  //i的第2^j祖先就是i的第2^(j-1)祖先的第2^(j-1)祖先
                //i的父辈们的fa数组都已经计算完毕,所以可以用来计算i
}
int lca(int u,int v)
{
    if(d[u]<d[v])
    swap(u,v);   //保证深度大的点为u,方便操作
    int dc=d[u]-d[v];   //计算深度差
    int i;
    for(i=0;i<=16;i++)  //值得注意的是,这里需要从零枚举
    {                    //此循环用于提到深度相同
        if((1<<i)& dc)   
            u=fa[u][i];
    }
    if(u==v)   //如果深度一样时,两个点相同,直接返回
        return u;
    for(i=16;i>=0;i--)  //如果从小到大的话就有可能无法提到正确位置
    {
        if(fa[u][i]!=fa[v][i])//跳2^j步,如果不一样,就跳,否则不跳 如果相等有可能都为0 即超过根节点
        {//在同一高度了,他们一起向上跳,跳他们不相同节点,当全都跳完之后fa[u][0]就是lca
            u=fa[u][i];
            v=fa[v][i];
        }
    }
    u=fa[u][0]; //上述过程做完,两点都在LCA下一层,所以走一步即可
    return u;
}
int main()
{
    int t;
    int x,y,z;
    scanf("%d",&t);
    for(int k=1;k<=t;k++)
    {
        init();
        scanf("%d%d",&n,&m);
        for(int i=0;i<=n;i++)
            g[i].clear();
        for(int i=1;i<n;i++)
        {
            scanf("%d%d%d",&x,&y,&z);
            g[x].push_back(node(y,z));
            fa[y][0]=x;
            if(fa[x][0]==0)    //查找根节点
                root=x;
        }
        d[root]=1;
        dfs(root);
        bz();
        int a,b;
        for(int i=0;i<m;i++)
        {
            scanf("%d%d",&a,&b);
            int ans=dis[a]+dis[b]-2*dis[lca(a,b)];
            printf("%d\n",ans);
        }
 
    }
    return 0;
}

 

  • 2
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值