hiho 1081 最短路径

复习最短路径算法，找了 hiho上的1081题；

算法解释见《算法导论》 P383.

#include<iostream>
#include<vector>

#include<climits>
using namespace std;

int main() {
	int N, M, S, T;
	cin >> N >> M >> S >> T;
	vector<int> tmp(N, -1);
	vector<vector<int>>w(N, tmp);
	int x, y, z;
	vector<int> d(N, INT_MAX);
	vector<bool> visit(N, false);
	for (int i = 0; i < M; ++i) { //路径的保存比较恶心，因为两个节点之间可能有多条路径
		cin >> x >> y >> z;
		if (w[x - 1][y - 1] != -1)
			w[x - 1][y - 1] = z > w[x - 1][y - 1] ? w[x - 1][y - 1] : z;
		else
			w[x - 1][y - 1] = z;
		if (w[y - 1][x - 1] != -1)
			w[y - 1][x - 1] = w[y - 1][x - 1] > w[x - 1][y - 1] ? w[x - 1][y - 1] : w[y - 1][x - 1];
		else
			w[y - 1][x - 1] = w[x - 1][y - 1];
		
	}
	d[S - 1] = 0; //初始化入口点的权值为 0
	int k = N;
	while (k--) { //每次一个节点，循环 N 次即可
		int min_val = INT_MAX, u;
		for (int i = 0; i < N; ++i) //找出权值最小的点，这里可以用优先级队列
			if (visit[i] == false && d[i] < min_val) {
				u = i;
				min_val = d[i];
			}
		visit[u] = true;//更新从节点 u 可以到达的点 的权值（第一轮 u即为入口点）
		for (int i = 0; i < N; ++i)// 
			if (visit[i] == false && w[u][i] != -1)
				d[i] = d[i] > (d[u] + w[u][i]) ? (d[u] + w[u][i]) : d[i];

	}
	cout << d[T - 1] << endl; //完成之后，d[i]即为 从源点到 i 节点的最短路径
	return 0;

}