hdu 1874 畅通工程续

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 <memory.h>
#define NUM 205
#define INF 0x3f3f3f3f
using namespace std;
int dis[NUM][NUM];
int g[NUM][NUM];
int n;
int m;
void Floyd()
{
    for(int i=0; i<n; i++)
    {
        for(int j=0; j<n; j++)
        {
            dis[i][j]=g[i][j];
        }
    }
    for(int k=0; k<n; k++)
    {
        for(int i=0; i<n; i++)
        {
            for(int j=0; j<n; j++)
            {
                if(dis[i][j]>dis[i][k]+dis[k][j])
                {
                    dis[i][j]=dis[i][k]+dis[k][j];
                }
            }
        }
    }
}
int main()
{
    int s,e,t;
    while(cin>>n>>m)
    {
        memset(g,INF,sizeof(g));
        for(int i=0;i<n;i++){     //必须加上赋值为0的操作，否则错误
            g[i][i]=0;
        }
        for(int i=0;i<m;i++)
        {
            cin>>s>>e>>t;
            if(g[s][e]>t)
            g[s][e]=g[e][s]=t;
        }
        Floyd();
        cin>>s>>e;
        if(dis[s][e]<INF)
            cout<<dis[s][e]<<endl;
        else
            cout<<"-1"<<endl;

    }
    return 0;
}