HDU 2874 Connections between cities（LCA+并查集）

 

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

 

【题目大意】

　　有n个村庄，m条路，不存在环，有q个询问，问两个村庄是否可达，
　　如果可达则输出最短路。

 

【题解】

　　因为不存在环，所以是森林，我们计算每个连通块的dfs序，计算块内每个点到根距离
　　当两个点在同一个连通块时，我们输出其dis值之和减去其LCA的dis值，
　　否则输出不想连。

 

【代码】

#include <cstdio>
#include <algorithm>
#include <climits>
#include <cstring> 
using namespace std;
const int N=300010;
int d[N],num[N],dis[N],ed=0,x,y,c,n,m,i,w[N],v[N],vis[N],f[N],g[N],nxt[N],size[N],son[N],st[N],en[N],dfn,top[N],t;char ch;
void read(int&a){char c;while(!(((c=getchar())>='0')&&(c<='9')));a=c-'0';while(((c=getchar())>='0')&&(c<='9'))(a*=10)+=c-'0';}
void add(int x,int y,int _w){v[++ed]=y;w[ed]=_w;nxt[ed]=g[x];g[x]=ed;}
void dfs(int x){
    size[x]=1;
    for(int i=g[x];i;i=nxt[i])if(v[i]!=f[x]){
        f[v[i]]=x,d[v[i]]=d[x]+1;dis[v[i]]=dis[x]+w[i];
        dfs(v[i]),size[x]+=size[v[i]];
        if(size[v[i]]>size[son[x]])son[x]=v[i];
    }
}
void dfs2(int x,int y){
    st[x]=++dfn;top[x]=y;
    if(son[x])dfs2(son[x],y);
    for(int i=g[x];i;i=nxt[i])if(v[i]!=son[x]&&v[i]!=f[x])dfs2(v[i],v[i]);
    en[x]=dfn;
}
int lca(int x,int y){
    for(;top[x]!=top[y];x=f[top[x]])if(d[top[x]]<d[top[y]]){int z=x;x=y;y=z;}
    return d[x]<d[y]?x:y;
}
void init(){ 
    memset(g,dfn=ed=0,sizeof(g));
    memset(v,0,sizeof(v));
    memset(nxt,0,sizeof(nxt));
    memset(son,-1,sizeof(son));
}
int fa[N],q;
int sf(int x){return fa[x]==x?x:fa[x]=sf(fa[x]);}
int main(){
    while(~scanf("%d%d%d",&n,&m,&q)){
        init();
        for(int i=1;i<=n;i++)fa[i]=i;
        for(int i=0;i<m;i++){
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            add(a,b,c); add(b,a,c);
            if(sf(a)!=sf(b)){
                fa[sf(a)]=sf(b);
            }
        }
        for(int i=1;i<=n;i++){
            if(fa[i]==i){
                dfs(i);
                dfs2(i,i);
            }
        } 
        while(q--){
            scanf("%d%d",&x,&y);
            if(sf(x)!=sf(y))puts("Not connected");
            else printf("%d\n",dis[x]+dis[y]-2*dis[lca(x,y)]);
        }
    }return 0;
}