【LCA模板】HDU - 2874 - Connections between cities

版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/monochrome00/article/details/82594623

题目链接<http://acm.hdu.edu.cn/showproblem.php?pid=2874>

---

题意：

给出一个森林，要求判断两个点是否在同一棵树上，如果在输出两者之间的简单路径长度。

---

题解：

关于LCA：https://www.cnblogs.com/FuTaimeng/p/5655616.html

是否在同一棵树上就用并查集判断一下。

树上两点间距就是根节点到两点间距之和，减去根节点到最近公共祖先距离的两倍。

---

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=1e4+7;

int fa[N][20],d[N],dis[N];
int p[N],edn;
int n,m,c;

struct Edge{
    int u,v,w,nxt;
    Edge(int u=0,int v=0,int w=0,int nxt=0):u(u),v(v),w(w),nxt(nxt){}
}edge[N*2];
void add(int u,int v,int w){
    edge[++edn]=Edge(u,v,w,p[u]);p[u]=edn;
    edge[++edn]=Edge(v,u,w,p[v]);p[v]=edn;
}

void dfs(int u){
    for(int i=p[u];~i;i=edge[i].nxt){
        int v=edge[i].v;
        if(v==fa[u][0])continue;
        d[v]=d[u]+1;
        dis[v]=dis[u]+edge[i].w;
        fa[v][0]=u;
        dfs(v);
    }
}
void init(){
    for(int j=1;(1<<j)<=n;j++){
        for(int i=1;i<=n;i++){
            fa[i][j]=fa[fa[i][j-1]][j-1];
        }
    }
}
int lca(int a,int b){
    if(d[a]>d[b])swap(a,b);
    int f=d[b]-d[a];
    for(int i=0;(1<<i)<=f;i++){
        if((1<<i)&f) b=fa[b][i];
    }

    if(a!=b){
        for(int i=(int)log2(N);i>=0;i--){
            if(fa[a][i]!=fa[b][i]){
                a=fa[a][i]; b=fa[b][i];
            }
        }
        a=fa[a][0];
    }
    return a;
}

int f[N];
int fd(int x){return x==f[x]?x:f[x]=fd(f[x]);}

int main()
{
    while(scanf("%d%d%d",&n,&m,&c)!=EOF){
        memset(p,-1,sizeof(p)),edn=-1;
        memset(d,0,sizeof(d));
        memset(dis,0,sizeof(dis));
        for(int i=1;i<=n;i++) f[i]=i;
        int u,v,w;
        for(int i=1;i<=m;i++){
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w);
            int fu=fd(u),fv=fd(v);
            f[fu]=fv;
        }
        for(int i=1;i<=n;i++){
            if(i==fd(i)) dfs(i);
        }
        init();
        for(int i=1;i<=c;i++){
            scanf("%d%d",&u,&v);
            int fu=fd(u),fv=fd(v);
            if(fu==fv){
                int pp=lca(u,v);
                printf("%d\n",dis[u]+dis[v]-2*dis[pp]);
            }
            else printf("Not connected\n");
        }
    }
}