hdu 1874 畅通工程续

思路：spfa算法

#include <iostream>
#include <stdio.h>
#include <cstring>
#include <queue>
#include <algorithm>
#define INF 0x3f3f3f3f
using namespace std;
int v[4000],u[4000],d[4000],first[4000],nex[4000],w[4000];
int vis[4000];
int e;
void add(int a,int b, int c)
{
    u[e]=a;
    v[e]=b;
    nex[e]=first[a];
    w[e]=c;
    first[a]=e++;
}
void spfa(int src)
{
    queue<int> q;
    memset(d,0x3f,sizeof(d));
    d[src]=0;
    vis[src]=1;
    q.push(src);
    while(!q.empty())
    {
        int u=q.front();
        vis[u]=0;
        q.pop();
        for(int i=first[u];i!=-1;i=nex[i])
        {
            if(d[v[i]]>d[u]+w[i])
            {
                d[v[i]]=d[u]+w[i];
                if(!vis[v[i]])
                q.push(v[i]);
                vis[v[i]]=1;
            }
        }
    }
}
int n,m;
int main()
{
   // freopen("in.txt","r",stdin);
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        e=0;
        int a,b,c;
        memset(u,0,sizeof(u));
        memset(v,0,sizeof(v));
        memset(vis,0,sizeof(vis));
        memset(first,-1,sizeof(first));
        memset(nex,-1,sizeof(nex));
        memset(w,0,sizeof(w));
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d",&a,&b,&c);
            add(a,b,c);
            add(b,a,c);
        }
        int x,y;
        scanf("%d%d",&x,&y);
        spfa(x);
        int ans=d[y];
        if(ans!=INF)
            printf("%d\n",d[y]);
        else
            printf("-1\n");
    }
}