HDU 2874

P 点击打开链接

#include<stdio.h>
#include<string.h>
#include<stack>
#include<string>
#include<math.h>
#include<queue>
#include<set>
#include<algorithm>
#include<iostream>
#include<vector>
#include<map>
using namespace std;
#define LL __int64
#define inf 1999999999
#define N 10010
int p[N],ehead[N],qhead[N],dis[N],f[N],ans[N*100];
int e,qe,n,m,cnt,Index;
bool vis[N],mark[N];
struct EDGE
{
    int v,next,w;
}edge[N*2];
struct Query
{
    int num,v,next;
}query[N*100*2];
void init()
{
    memset(ehead,-1,sizeof(ehead));
    memset(qhead,-1,sizeof(qhead));
    memset(dis,0,sizeof(dis));
    memset(mark,0,sizeof(mark));
    memset(ans,-1,sizeof(ans));
    for(int i=1;i<=n;i++)
        p[i]=f[i]=i;
    e=qe=cnt=Index=0;
}
void adde(int u,int v,int w)
{
    edge[e].v=v;
    edge[e].w=w;
    edge[e].next=ehead[u];
    ehead[u]=e++;
}
void addq(int u,int v,int num)
{
    query[qe].v=v;
    query[qe].num=num;
    query[qe].next=qhead[u];
    qhead[u]=qe++;
}
int find(int x)
{
    if(x!=f[x]) f[x]=find(f[x]);
    return f[x];
}
void Union(int u,int v)
{
    int fu=find(u);
    int fv=find(v);
    if(fv!=fu) f[fv]=fu;
}
void tarjan(int u,int d)
{
    int i,v;
    mark[u]=vis[u]=1;dis[u]=d;
    for(i=ehead[u];i!=-1;i=edge[i].next)
    {
        v=edge[i].v;
        if(vis[v]) continue;
        tarjan(v,d+edge[i].w);
        Union(u,v);
    }
    for(i=qhead[u];i!=-1;i=query[i].next)
    {
        v=query[i].v;
        if(vis[v])
            ans[query[i].num]=dis[u]+dis[v]-2*dis[find(v)];
    }
}
int main()
{
    int i,j,u,v,w,q;
    while(scanf("%d%d%d",&n,&m,&q)!=-1)
    {
        init();
        for(i=1;i<=m;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            adde(u,v,w);
            adde(v,u,w);
        }
        for(i=1;i<=q;i++)
        {
            scanf("%d%d",&u,&v);
            addq(u,v,i);
            addq(v,u,i);
        }
        for(i=1;i<=n;i++)
        {
            if(!mark[i])
            {
                memset(vis,0,sizeof(vis));
                tarjan(i,0);
            }
        }
        for(i=1;i<=q;i++)
        {
            if(ans[i]>=0)
                printf("%d\n",ans[i]);
            else puts("Not connected");
        }
    }
    return 0;
}