hdu 1874 畅通工程续（SPFA模板）

畅通工程续

Time Limit: 3000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 82186    Accepted Submission(s): 31619

Problem Description

某省自从实行了很多年的畅通工程计划后，终于修建了很多路。不过路多了也不好，每次要从一个城镇到另一个城镇时，都有许多种道路方案可以选择，而某些方案要比另一些方案行走的距离要短很多。这让行人很困扰。

现在，已知起点和终点，请你计算出要从起点到终点，最短需要行走多少距离。

 

Input

本题目包含多组数据，请处理到文件结束。
每组数据第一行包含两个正整数N和M(0<N<200,0<M<1000)，分别代表现有城镇的数目和已修建的道路的数目。城镇分别以0～N-1编号。
接下来是M行道路信息。每一行有三个整数A,B,X(0<=A,B<N,A!=B,0<X<10000),表示城镇A和城镇B之间有一条长度为X的双向道路。
再接下一行有两个整数S,T(0<=S,T<N)，分别代表起点和终点。

 

Output

对于每组数据，请在一行里输出最短需要行走的距离。如果不存在从S到T的路线，就输出-1.

 

Sample Input

3 3

0 1 1

0 2 3

1 2 1

0 2

3 1

0 1 1

1 2

 

Sample Output

2

-1

 

#include<iostream>
#include<string.h>
#include<string>
#include<algorithm>
#include<queue>
#define ll long long
#define mx 0x3f3f3f3f
using namespace std;
int way[][],dis[],vis[],cnt[];
//way记录路径关系，dis[i]记录起点到点j的最近距离，vis[i]标记点是否在队列中，cnt[i]记录点i进入队列的次数
int n,m;
void init()
{
    for(int i=;i<n;i++)//先初始化way
    {
        for(int j=;j<n;j++)
        {
            if(i==j)
                way[i][j]=;
            else
                way[i][j]=mx;
        }
    }
}

void spfa(int st)
{
    for(int i=;i<n;i++)//这里点编号是从0开始的
        dis[i]=mx;
    memset(vis,,sizeof(vis));
    memset(cnt,,sizeof(cnt));
    vis[st]=;
    cnt[st]=;
    dis[st]=;
    queue<int>p;
    p.push(st);
    while(!p.empty())
    {
        int now=p.front();
        p.pop();
        vis[now]=;
        for(int i=;i<n;i++)
        {
            if(dis[now]+way[now][i]<dis[i])
            {
                dis[i]=dis[now]+way[now][i];
                if(vis[i]==)//如果点不在队列里面
                {
                    p.push(i);
                    vis[i]=;
                    cnt[i]++;
                    if(cnt[i]>n)//如果这个点加入超过n次，说明存在负圈，直接返回
                        return ;
                }
            }
        }
    }

}
int main()
{
    int st,end;
    while(~scanf("%d%d",&n,&m))//输入到结束，否则TLE
    {
        init();//初始化way
        for(int i=;i<m;i++)
        {
            int x,y,z;
            scanf("%d%d%d",&x,&y,&z);
            if(way[x][y]>z)//重边
            {
                way[x][y]=z;
                way[y][x]=z;
            }

        }
        scanf("%d%d",&st,&end);
        spfa(st);//spfa可以求出起点到每一个点的最短距离
        if(dis[end]==mx)
            printf("-1\n");
        else
            printf("%d\n",dis[end]);
    }
    return ;
}