畅通工程续—杭电OJ

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 <bits/stdc++.h>
using namespace std;
#define INF 0x3f3f3f3f
const int N = 205;
int e[N][N];
int dis[N];
bool vis[N];
int n, m, st, ed;
//初始化整个e数组，一个长度和宽度都为N的矩阵，让每个城镇到另外的城镇的距离都为INF
inline void init()
{
    for(int i = 0; i < n; ++i)
        for(int j = 0; j < n; ++j)
            e[i][j] = ( i==j ? 0 : INF);
}
//算法核心
void Dijkstra()
{
    int k;
    for(int i = 0; i < n - 1; ++i)
    {
        int minv = INF;
        for(int j = 0; j < n; ++j)
            if(dis[j] < minv && vis[j]==0){ //表示源点找到离它最近的那个点k vis[j]==0 意思选那些个之前没被操作的点
                minv = dis[j];
                k = j;//记录点k的位置
        }
        vis[k] = true; //标记下k点，使得接下来再找离k点最近的点时跳过k点本身
        for(int v = 0; v < n; ++v)
        {
            if(e[k][v] < INF )//表示k点能够到达的点
                dis[v] = min(dis[v],dis[k]+e[k][v]);//就把起点到k点能到达的点v的距离更新为起点直接到v点和起点先到达k点再到达v点当中更小的那个
        }
    }
}
int main()
{
    int a, b, c;
    while(cin >> n >> m)
    {
        init();
        for(int i = 0; i < m; ++i)
        {
            cin >> a >> b >> c;
            if(e[a][b] > c)//排除了a = b，也就是只要a!=b,那么这个距离就合法，就可以让a b 之间的双向距离赋值为c
                e[a][b] = e[b][a] = c;
        }
        cin >> st >> ed;
        //初始化dis数组，也就是表示起点到其余点最短距离的数组
        for(int i = 0; i < n; ++i)
            dis[i] = e[st][i];
        //初始化vis数组
        memset(vis, 0, sizeof(vis));
        vis[st] = 1;//使得在接下来找离起点最近的点时跳过起点
        Dijkstra();
        cout <<(dis[ed]==INF ? -1 : dis[ed]) <<endl;
    }
    return 0;
}