【HDU 2874】Connections between cities（LCA）

dfs找出所有节点所在树及到树根的距离及深度及父亲。

i和j在一棵树上，则最短路为dis[i]+dis[j]-dis[LCA(i,j)]*2。

#include <cstring>
#include <cstdio>
#define N 10005
#define add(u,v,w) e[++cnt]=(edge){v,head[u],w};head[u]=cnt
using namespace std;
struct edge{
    int to,next,w;
}e[N<<];
int head[N],cnt,f[N],dis[N],deep[N],tr[N],t;
int fa(int i,int j){
    while(i!=j)
        if(deep[i]>deep[j])
            i=f[i];
        else
            j=f[j];
    return i;
}
void dfs(int x,int p){
    tr[x]=t;
    deep[x]=deep[p]+;
    f[x]=p;
    for(int i=head[x];i;i=e[i].next){
        int v=e[i].to;
        if(v==p)continue;
        dis[v]=dis[x]+e[i].w;
        dfs(v,x);
    }
}
int main(){
    int n,m,c,u,v,w;
    while(~scanf("%d%d%d",&n,&m,&c)){
        memset(head,,sizeof head);
        memset(f,,sizeof f);
        cnt=t=;
        for(int i=;i<=m;i++){
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w);
            add(v,u,w);
        }
        for(int i=;i<=n;i++)
            if(f[i]==){
                t++;
                dfs(i,);
            }
        for(int i=;i<=c;i++){
            scanf("%d%d",&u,&v);
            if(tr[u]!=tr[v])
                puts("Not connected");
            else
                printf("%d\n",dis[u]+dis[v]-dis[fa(u,v)]*);
        }
    }
}