【HDU 2874】Connections between cities（LCA）

#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<<1];
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]+1;
    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,0,sizeof head);
        memset(f,0,sizeof f);
        cnt=t=0;
        for(int i=1;i<=m;i++){
            scanf("%d%d%d",&u,&v,&w);
            add(u,v,w);
            add(v,u,w);
        }
        for(int i=1;i<=n;i++)
            if(f[i]==0){
                t++;
                dfs(i,0);
            }
        for(int i=1;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)]*2);
        }
    }
}