ZCMU 1398(spfa最短路径)

1398: 工程

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

SPFA
Code:

#include <stdio.h>
#include<queue>
#include<algorithm>
#include<iostream>
#define inf 0x3f3f3f3f
using namespace std;
int map[205][205];
int vis[205];
int dis[205];
int n,m;
void init()
{
    for(int i=0; i<n; i++)
        for(int j=0; j<n; j++)
        {
            if(i==j)
                map[i][j]=0;
            else
                map[i][j]=inf;
        }
    for(int i=0; i<n; i++)
    {
        vis[i]=0;
        dis[i]=inf;
    }

}
int SPFA(int s,int e)
{
    dis[s]=0;
    queue<int>q;
    q.push(s);
    vis[s]=1;
    while(!q.empty())
    {
        int k=q.front();
        q.pop();
        vis[k]=0;
        for(int i=0; i<n; i++)
        {
            if(dis[i]>dis[k]+map[k][i])
            {
                dis[i]=dis[k]+map[k][i];
                if(vis[i]==0)
                {
                    q.push(i);
                    vis[i]=1;
                }
            }
        }
    }
    return dis[e];
}
int main()
{

    while(scanf("%d %d",&n,&m)!=EOF)
    {
        init();
        for(int i=0; i<m; i++)
        {
            int a,b,c;
            scanf("%d %d %d",&a,&b,&c);
            map[a][b]=c;
            map[b][a]=c;
        }
        int s,t;
        scanf("%d %d",&s,&t);
        int ans=SPFA(s,t);

        if(ans!=inf)
            printf("%d\n",ans);
        else
            printf("-1\n");
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/weixin_44941429/article/details/91657007