最短路径简单应用 畅通工程续 POJ 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 <string.h>
using namespace std;

#define INF 1000000
int map[200][200];
bool visited[200];
int disit[200]; //存储距离
int N, M;
//s开始位置，e结束位置
void dij(int s, int e)
{
	memset(visited, false, sizeof(visited));
	for (int i = 0; i < N; i++)
	{
		if (map[s][i] == -1)
			disit[i] = INF;
		else
			disit[i] = map[s][i];
	}
	visited[s] = true;
	for (int i = 1; i < N; i++)
	{
		int min = INF;
		int index = 0;
		for (int j = 0; j < N; j++)
		{
			if (!visited[j] && j != s && disit[j] != -1 && disit[j] < min)
			{
				min = disit[j];
				index = j;
			}
		}
		visited[index] = true;
		if (index == s) //如果已经达到终点
			break;
		for (int j = 0; j < N; j++)
		{
			if (!visited[j] && map[index][j] != -1
					&& (min + map[index][j]) < disit[j])
			{
				disit[j] = min + map[index][j];
			}
		}
	}
}

int main()
{
	int a, b, c;
	int s, e;
	while (cin >> N >> M)
	{
		memset(map, -1, sizeof(map)); //-1表示不可达
		while (M--)
		{
			cin >> a >> b >> c;
			//有可能有更小的路径
			if (map[a][b] == -1 || c < map[a][b])
			{
				map[a][b] = c;
				map[b][a] = c;
			}
		}
		cin >> s >> e;

		if (s == e)
		{
			cout << 0 << endl;
			continue;
		}
		dij(s, e);
		if (visited[e] && disit[e] < INF)
			cout << disit[e] << endl;
		else
			cout << -1 << endl;
	}
	return 0;
}