hiho 1081 最短路径

#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;

}