hdu2874（tarjan）

题目链接：HDU - 2874

比较少写tarjan，还要多练习。。。

#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int maxn=10010;
const int maxq=1000010;
int n,m,k;
int u,v,w;
struct edge
{
    int v,w;
    int nex;
}e[maxn*2];
struct query
{
    int v;
    int id;
    int nex;
}q[maxq*2];
int ect,qct;
int ehead[maxn];
int qhead[maxn];
int f[maxn],id[maxn],dis[maxn],vis[maxn];
int ans[maxq];
void adde(int u,int v,int w)
{
    e[ect].v=v;
    e[ect].w=w;
    e[ect].nex=ehead[u];
    ehead[u]=ect++;
}
void addq(int u,int v,int id)
{
    q[qct].v=v;
    q[qct].id=id;
    q[qct].nex=qhead[u];
    qhead[u]=qct++;
}
int gf(int x)
{
    return x==f[x]?x:f[x]=gf(f[x]);
}

void init()
{
    qct=ect=0;
    memset(qhead,-1,sizeof(qhead));
    memset(ehead,-1,sizeof(ehead));
    memset(vis,0,sizeof(vis));
}

void tarjan(int u,int tp)
{
    vis[u]=1;
    f[u]=u;
    id[u]=tp;
    for(int i=qhead[u];i!=-1;i=q[i].nex)
    {
        int v=q[i].v;
        if(vis[v])
        {
            if(id[v]==id[u])
                ans[q[i].id]=dis[u]+dis[v]-2*dis[gf(v)];
            else
                ans[q[i].id]=-1;
        }
    }
    for(int i=ehead[u];i!=-1;i=e[i].nex)
    {
        int v=e[i].v;
        if(!vis[v])
        {
            dis[v]=dis[u]+e[i].w;
            tarjan(v,tp);
            f[v]=u;
        }
    }
}

int main()
{
    int n,m,k;
    while(scanf("%d%d%d",&n,&m,&k)!=EOF)
    {
        init();
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&u,&v,&w);
            adde(u,v,w);
            adde(v,u,w);
        }
        for(int i=0;i<k;i++)
        {
            scanf("%d%d",&u,&v);
            addq(u,v,i);
            addq(v,u,i);
        }
        for(int i=1;i<=n;i++)
            if(!vis[i])
            {
                dis[i]=0;
                tarjan(i,i);
            }
        for(int i=0;i<k;i++)
        {
            if(ans[i]==-1) puts("Not connected");
            else printf("%d
",ans[i]);
        }
    }
}