HDU 2874 Connections between cities（LCA Tarjan）

Connections between cities

题意：求连接两城市的最短距离

题目思路:LCA Tarjan

解题方法:通过Tarjan求距离

需注意的问题：存在森林，所以对于每棵树都得进行计算

    存在未连通，只需先判断节点的根节点（并查集find）是否相同即可

    内存 29432K险过，需用邻接表模拟，vector会爆

AC代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<vector>
using namespace std;
#define N 10010
#define NN 2000010
struct Edge
{
    int v,len,next;
} edge[N*2];
int tot,head[N];
void add(int a,int b,int c)
{
    edge[tot].v=b;
    edge[tot].len=c;
    edge[tot].next=head[a];
    head[a]=tot++;

    edge[tot].v=a;
    edge[tot].len=c;
    edge[tot].next=head[b];
    head[b]=tot++;
}
struct Query{
    int v,index,next;
};
Query Q[NN];
int q_tot,q_head[N];
void add_ques(int a,int b,int index)
{
    Q[q_tot].v=b;
    Q[q_tot].index=index;
    Q[q_tot].next=q_head[a];
    q_head[a]=q_tot++;

    Q[q_tot].v=a;
    Q[q_tot].index=index;
    Q[q_tot].next=q_head[b];
    q_head[b]=q_tot++;
}
bool vis[N],root[N];
int pre[N],dis[N],ans[NN/2];
int n,q;
void init(){
    for(int i=0;i<=n;i++){
        pre[i]=i;
        vis[i]=false;
    }
    tot=q_tot=0;
    memset(q_head,-1,sizeof(q_head));
    memset(head,-1,sizeof(head));
    memset(ans,-1,sizeof(ans));
    memset(root,true,sizeof(root));
}
int find(int x){
    return pre[x]==x?pre[x]:pre[x]=find(pre[x]);
}
void Union(int x,int y){
    x=find(x);
    y=find(y);
    if(x!=y){
        pre[y]=x;
    }
}
void Tarjan(int cur,int val){
    dis[cur]=val;
    vis[cur]=true;
    for(int j=head[cur]; j!=-1; j=edge[j].next)
    {
        int v=edge[j].v;
        if(vis[v])    continue;
        Tarjan(v,val+edge[j].len);
        Union(cur,v);
    }
    for(int j=q_head[cur]; j!=-1; j=Q[j].next)
    {
        int v=Q[j].v;
        if(!vis[v])    continue;
        ans[Q[j].index]=dis[v]+dis[cur]-2*dis[find(v)];
    }
}
int main(){
    int T,m;
   // scanf("%d",&T);
    while(scanf("%d%d%d",&n,&m,&q)!=EOF){
        int a,b,c;
        init();
        for(int i=0;i<m;i++){
            scanf("%d%d%d",&a,&b,&c);
            add(a,b,c);
            root[b]=false;
        }
        for(int i=0;i<q;i++){
            scanf("%d%d",&a,&b);
            add_ques(a,b,i);

        }
        ///对于森林的每颗树进行计算
        for(int i=1;i<=n;i++){
            if(!vis[i]&&root[i]){
                Tarjan(i,0);
            }
        }
        for(int i=0;i<q;i++){
            ///查找两节点的根节点，如果不相同则未连同
            if(find(Q[i*2].v)!=find(Q[2*i+1].v)) printf("Not connected\n");
            else printf("%d\n",ans[i]);
        }
    }
    return 0;
}