LCA

LCA最近公共祖先

学习链接

[https://blog.csdn.net/creatorx/article/details/70739834]

hdu 2586

[http://acm.hdu.edu.cn/showproblem.php?pid=2586]

题意

给出每两个相邻节点的距离
询问任意两点的距离

分析

Tarjan离线查询
先求出查询的lca，答案就是
d[s[i]]+d[e[i]]-2*d[lca[i]]

代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;

const int N=4e4 + 5;
const int M=210;
int head[N],s[M],e[M],d[N],lca[M],fa[N];
bool vis[N];

struct str{
    int to,next,dis;
}edg[N<<1];

int tot=0,n,m;
int find(int x)
{
    if(x==fa[x]) return x;
    return fa[x]=find(fa[x]);
}

void add(int x,int y,int z)
{
    int cur=++tot;
    edg[cur].next=head[x];
    head[x]=cur;
    edg[cur].to=y;
    edg[cur].dis=z;
}

void init()
{
    tot=0;
    memset(vis,0,sizeof(vis));
    memset(head,0,sizeof(head));
    memset(d,0,sizeof(d));
    for(int i=1;i<=n;i++) fa[i]=i;
}
void tarjan(int u)
{
    vis[u]=1;
    for(int i=1;i<=m;i++)
    {
        if(s[i]==u&&vis[e[i]])
        lca[i]=find(e[i]);
        if(e[i]==u&&vis[s[i]])
        lca[i]=find(s[i]);
    }
    for(int i=head[u];i;i=edg[i].next)
    {
        int cur=edg[i].to;
        if(!vis[cur])
        {
            d[cur]=d[u]+edg[i].dis;
            tarjan(cur);
            fa[cur]=u;
        }
    }
}

int main()
{
    int t;
    scanf("%d",&t);
    while(t--)
    {
        scanf("%d%d",&n,&m); 
        init();
        int u,v,w;
        for(int i=1;i<n;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w);
            add(v,u,w);
        } 
        for(int i=1;i<=m;i++)
         scanf("%d%d",&s[i],&e[i]);
     
        tarjan(1);
        for(int i=1;i<=m;i++)
        printf("%d\n",d[s[i]]+d[e[i]]-2*d[lca[i]]);
    } 
    return 0;
 }