畅销工程续
Dijkstra算法的基本应用
题目
畅通工程续
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
代码
//思路 : dijkstra 算法求解单源点最短路径
//无向图,有权值
#include <iostream>
#include <algorithm>
#include <cstring>
using namespace std;
int inf = 888888;
int N,M;
int a,b,x;
int T ,S;
int map[100][100];
int dis[100];
int vis[100];
int main(int argc, char** argv) {
while(cin >>N>>M)
{
//初始化
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 < M; i++)
{
cin >> a >> b >> x;
map[a][b] = map[b][a] = x;
}
cin >> S >>T; //出发城市,目标城市
//初始化dis
for(int i = 0; i <= N; i++)
{
dis[i] = inf;
}
//初始化vis
memset(vis,0,sizeof(vis));
vis[S] = 1;
//第一次更新 dis
for(int j = 0; j <= N; j++)
{
dis[j] = map[S][j];
}
//循环遍历
for(int i = 0; i < N; i++)
{
int temp ; //标记最短距离的源点
int minn = inf; //最短距离
//找到更新之前的最短路径源点,并标记
for(int j = 0; j < N; j++)
{
if(!vis[j] && dis[j] < minn)
{
minn = dis[j];
temp = j;
}
}
vis[temp] = 1; //存放
//根据找到的最短距离源点更新dis
for(int j = 0 ; j < N; j++)
{
if(!(vis[j]) && (dis[j] > dis[temp]+map[temp][j]))
{
dis[j] = dis[temp] + map[temp][j];
}
}
}
if(dis[T] != inf) cout << dis[T] << endl;
else cout << "-1" << endl;
}
return 0;
}